HMPhuoc commited on
Commit
3c61a05
1 Parent(s): 7163957

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,11 +21,11 @@ from underthesea import word_tokenize
21
  # tokenizer = pickle.load(f)
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
 
 
21
  # tokenizer = pickle.load(f)
22
 
23
  #Load LSTM
24
+ fp = Path(__file__).with_name('lstm_model.h5')
25
  LSTM_model = tf.keras.models.load_model(fp)
26
 
27
  #Load GRU
28
+ fp = Path(__file__).with_name('gru_model.h5')
29
  GRU_model = tf.keras.models.load_model(fp)
30
 
31