Update app.py

app.py

@@ -15,23 +15,27 @@ import unicodedata as ud
 from underthesea import word_tokenize
 
 from phoBERT import BERT_predict
-
+import time
 
 LSTM_model = tf.keras.models.load_model('lstm_model.tf')
 
 GRU_model = tf.keras.models.load_model('gru_model.tf')
 
 def LSTM_predict(x):
-    
+    t1 = time.time()
     pred_proba = LSTM_model.predict([x])[0]
-
+    t2 = time.time()
+    print(f'{t2-t1}s')
+    
     pred_proba = [round(i,2) for i in pred_proba]
 
     return pred_proba
 
 def GRU_predict(x):
-    
+    t1 = time.time()
     pred_proba = GRU_model.predict([x])[0]
+    t2 = time.time()
+    print(f'{t2-t1}s')
 
     pred_proba = [round(i,2) for i in pred_proba]