Nguyen Thi Dieu Hien commited on
Commit
f8acede
·
unverified ·
1 Parent(s): 587adfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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