add judgePlus

app.py

@@ -76,7 +76,7 @@ def plot(result):
   return p
   pass
 
-def judge(x, bert=False):
+def judge(x):
 
   label = ['độc hại', 'cực kì độc hại', 'tục tĩu', 'đe dọa', 'xúc phạm', 'thù ghét cá nhân']
   result = []
@@ -93,13 +93,33 @@ def judge(x, bert=False):
   result_lstm = np.round(lstm_pred, 2)
   result_gru = np.round(gru_pred, 2)
   #result_bert = np.round(bert_pred, 2)
-  if bert == True:
-    bert_pred = BERT_predict(x)
-    result_bert = np.round(bert_pred, 2)
-    for i in range(6):
-      result.append((result_lstm[i]+result_gru[i]+result_bert[i])/3)
-  else:
-    for i in range(6):
-      result.append((result_lstm[i]+result_gru[i])/2)
+
+  for i in range(6):
+    result.append((result_lstm[i]+result_gru[i])/2)
+  
+  return (result)
+
+def judgePlus(x):
+
+  label = ['độc hại', 'cực kì độc hại', 'tục tĩu', 'đe dọa', 'xúc phạm', 'thù ghét cá nhân']
+  result = []
+  judge_result = []
+
+  x = ud.normalize('NFKC', 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)
   
   return (result)

main.py

@@ -1,5 +1,5 @@
 from flask import Flask, request
-from app import judge
+from app import judge, judgePlus
 from flask_cors import CORS
 import threading
 
@@ -27,7 +27,7 @@ def check():
 @app.route("/checkplus", methods=["POST"])
 def checkPlus():
     comment = request.json['comment']
-    result = judge(comment)
+    result = judgePlus(comment)
     return result
 
 if __name__ == '__main__':