add judgeBert & checkBert

app.py

@@ -127,3 +127,28 @@ def judgePlus(x):
     result.append((result_lstm[i]+result_gru[i]+result_bert[i])/3)
   
   return (result)
+
+def judgeBert(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)
+  x = word_tokenize(x, format="text")
+
+  
+  try:
+    bert_pred = BERT_predict(x)
+  except:
+    bert_pred = [0,0,0,0,0,0]
+  
+  
+  return_result = 'Result'
+  
+  result_bert = np.round(bert_pred, 2)
+  #result_bert = np.round(bert_pred, 2)
+  
+  
+  
+  return (result_bert)

main.py

@@ -1,5 +1,5 @@
 from flask import Flask, request
-from app import judge, judgePlus
+from app import judge, judgePlus, judgeBert
 from flask_cors import CORS
 import threading
 
@@ -30,6 +30,12 @@ def checkPlus():
     result = judgePlus(comment)
     return result
 
+@app.route("/checkbert", methods=["POST"])
+def checkBert():
+    comment = request.json['comment']
+    result = judgeBert(comment)
+    return result
+
 if __name__ == '__main__':
     
     app.run(debug=False, threaded=True)