Hugging Face's logo

Spaces:
HMPhuoc
/
toxic
Running

App Files Community
HMPhuoc commited on
Commit
33d20ba
•
1 Parent(s): 9c80df0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,11 +22,11 @@ from phoBERT import BERT_predict
22
 
23
  #Load LSTM
24
  fp = Path(__file__).with_name('lstm_model.keras')
25
- LSTM_model = keras.saving.load_model(fp)
26
 
27
  #Load GRU
28
  fp = Path(__file__).with_name('gru_model.keras')
29
- GRU_model = keras.saving.load_model(fp)
30
 
31
 
32
  def tokenizer_pad(tokenizer,comment_text,max_length=200):
 
22
 
23
  #Load LSTM
24
  fp = Path(__file__).with_name('lstm_model.keras')
25
+ LSTM_model = tf.keras.models.load_model(fp)
26
 
27
  #Load GRU
28
  fp = Path(__file__).with_name('gru_model.keras')
29
+ GRU_model = tf.keras.models.load_model(fp)
30
 
31
 
32
  def tokenizer_pad(tokenizer,comment_text,max_length=200):