Spaces:
Runtime error
Runtime error
Nguyen Thi Dieu Hien
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -94,6 +94,7 @@ def load_models(model_type):
|
|
| 94 |
elif model_type == 'bilstm_phobertbase':
|
| 95 |
model = load_model("models/bilstm_phobertbase.h5", compile=False)
|
| 96 |
tokenizer = AutoTokenizer.from_pretrained("vinai/phobert-base-v2")
|
|
|
|
| 97 |
max_len = 256
|
| 98 |
else:
|
| 99 |
raise ValueError("Invalid model type specified.")
|
|
@@ -101,7 +102,7 @@ def load_models(model_type):
|
|
| 101 |
if models is not None:
|
| 102 |
return models, tokenizer, max_len
|
| 103 |
else:
|
| 104 |
-
return model, tokenizer, max_len
|
| 105 |
|
| 106 |
|
| 107 |
|
|
|
|
| 94 |
elif model_type == 'bilstm_phobertbase':
|
| 95 |
model = load_model("models/bilstm_phobertbase.h5", compile=False)
|
| 96 |
tokenizer = AutoTokenizer.from_pretrained("vinai/phobert-base-v2")
|
| 97 |
+
phobert = AutoModel.from_pretrained("vinai/phobert-base-v2")
|
| 98 |
max_len = 256
|
| 99 |
else:
|
| 100 |
raise ValueError("Invalid model type specified.")
|
|
|
|
| 102 |
if models is not None:
|
| 103 |
return models, tokenizer, max_len
|
| 104 |
else:
|
| 105 |
+
return model, tokenizer, max_len, phobert
|
| 106 |
|
| 107 |
|
| 108 |
|