change models keras

app.py

@@ -16,16 +16,16 @@ from underthesea import word_tokenize
 from phoBERT import BERT_predict
 
 #Load tokenizer
-fp = Path(__file__).with_name('tokenizer.pkl')
-with open(fp,mode="rb") as f:
-    tokenizer = pickle.load(f)
+# fp = Path(__file__).with_name('tokenizer.pkl')
+# with open(fp,mode="rb") as f:
+#     tokenizer = pickle.load(f)
 
 #Load LSTM
-fp = Path(__file__).with_name('lstm_model.h5')
+fp = Path(__file__).with_name('lstm_model.keras')
 LSTM_model = load_model(fp)
 
 #Load GRU
-fp = Path(__file__).with_name('gru_model.h5')
+fp = Path(__file__).with_name('gru_model.keras')
 GRU_model = load_model(fp)
 
 
@@ -40,7 +40,7 @@ def tokenizer_pad(tokenizer,comment_text,max_length=200):
     return padded_sequences
 
 def LSTM_predict(x):
-    x = tokenizer_pad(tokenizer=tokenizer,comment_text=x)
+    #x = tokenizer_pad(tokenizer=tokenizer,comment_text=x)
 
     pred_proba = LSTM_model.predict(x)[0]
 
@@ -51,7 +51,7 @@ def LSTM_predict(x):
     return pred_proba
 
 def GRU_predict(x):
-    x = tokenizer_pad(tokenizer=tokenizer,comment_text=x)
+    #x = tokenizer_pad(tokenizer=tokenizer,comment_text=x)
 
     
     pred_proba = GRU_model.predict(x)[0]