HMPhuoc commited on
Commit
0b15904
1 Parent(s): 0b02c8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -21,12 +21,12 @@ with open(fp,mode="rb") as f:
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
 
32
  def tokenizer_pad(tokenizer,comment_text,max_length=200):
 
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('lstm_model.tf')
26
 
27
  #Load GRU
28
+ #fp = Path(__file__).with_name('gru_model.h5')
29
+ GRU_model = tf.keras.models.load_model('gru_model.tf')
30
 
31
 
32
  def tokenizer_pad(tokenizer,comment_text,max_length=200):