Nguyen Thi Dieu Hien commited on
Commit
a5846ab
·
unverified ·
1 Parent(s): bd5481a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +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
- phobert = AutoModel.from_pretrained("vinai/phobert-base-v2")
98
  max_len = 256
99
  else:
100
  raise ValueError("Invalid model type specified.")
 
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.")