remove bert from prediction

app.py

@@ -14,7 +14,7 @@ import unicodedata as ud
 
 from underthesea import word_tokenize
 
-from phoBERT import BERT_predict
+#from phoBERT import BERT_predict
 
 # Load tokenizer
 # fp = Path(__file__).with_name('tokenizer.pkl')
@@ -86,14 +86,14 @@ def judge(x):
 
   lstm_pred = LSTM_predict(x)
   gru_pred = GRU_predict(x)
-  bert_pred = BERT_predict(x)
+  #bert_pred = BERT_predict(x)
   #print(result)
   
   return_result = 'Result'
   result_lstm = np.round(lstm_pred, 2)
   result_gru = np.round(gru_pred, 2)
-  result_bert = np.round(bert_pred, 2)
+  #result_bert = np.round(bert_pred, 2)
   for i in range(6):
-    result.append((result_lstm[i]+result_gru[i]+result_bert[i])/3)
+    result.append((result_lstm[i]+result_gru[i])/2)
   
   return (result)

phoBERT.py

@@ -78,7 +78,14 @@ def BERT_predict(text):
     token_type_ids = token['token_type_ids']
     
     result = model(ids, mask, token_type_ids)
-    print(result)
+    # print(result)
     return result.tolist()[0]
 
-print(BERT_predict("xin chaof"))
+print(BERT_predict("xin chaof"))
+print(BERT_predict("con chó"))
+print(BERT_predict("đồ chó"))
+print(BERT_predict("đồ ngu"))
+print(BERT_predict("cái lồn"))
+print(BERT_predict("óc chó"))
+print(BERT_predict("đồ chó đẻ"))
+print(BERT_predict("con đĩ"))