Update app.py
Browse files
app.py
CHANGED
@@ -85,7 +85,7 @@ def loadModel():
|
|
85 |
# )
|
86 |
tokenizer = AutoTokenizer.from_pretrained(base_model,use_fast=False,cache_dir=cache_dir)
|
87 |
tokenizer.pad_token = tokenizer.unk_token
|
88 |
-
return model
|
89 |
|
90 |
model, tokenizer = loadModel()
|
91 |
|
|
|
85 |
# )
|
86 |
tokenizer = AutoTokenizer.from_pretrained(base_model,use_fast=False,cache_dir=cache_dir)
|
87 |
tokenizer.pad_token = tokenizer.unk_token
|
88 |
+
return model, tokenizer
|
89 |
|
90 |
model, tokenizer = loadModel()
|
91 |
|