fix bert complete

app.py

@@ -114,9 +114,8 @@ def judgePlus(x):
   try:
     bert_pred = BERT_predict(x)
   except:
-    bert_pred = [0.5, 0.5, 0.5, 0.5, 0.5, 0.5]
-  #bert_pred = BERT_predict(x)
-  #print(result)
+    bert_pred = np.average([lstm_pred, gru_pred], axis=0)
+  
   
   return_result = 'Result'
   result_lstm = np.round(lstm_pred, 2)

phoBERT.py

@@ -78,7 +78,6 @@ 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]