Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ def load_model(model_name):
|
|
44 |
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
|
45 |
tokenizer.pad_token = tokenizer.eos_token
|
46 |
|
47 |
-
print(f"Model Loading Time : {time.time() - start_time}.")
|
48 |
logger.info(f"Model Loading Time : {time.time() - start_time} .")
|
49 |
|
50 |
|
|
|
44 |
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
|
45 |
tokenizer.pad_token = tokenizer.eos_token
|
46 |
|
47 |
+
print(f"Model Loading Time : {time.time() - start_time}.")
|
48 |
logger.info(f"Model Loading Time : {time.time() - start_time} .")
|
49 |
|
50 |
|