Update main.py

main.py

@@ -1,8 +1,11 @@
 from flask import Flask, request
-from app import judge, judgePlus, judgeBert
+from app import judge, judgePlus, judgeBert, tokenize, lstm_predict, gru_predict
+from phoBERT import BERT_predict
 from flask_cors import CORS
 import threading
 
+import time
+
 app = Flask(__name__)
 CORS(app)
 
@@ -36,6 +39,107 @@ def checkBert():
     result = judgeBert(comment)
     return result
 
+@app.route("/check_time")
+def time():
+    l10 = []
+    l100 = []
+    l200 = []
+
+    t_lstm_10 = []
+    t_lstm_100 = []
+    t_lstm_200 = []
+
+    t_gru_10 = []
+    t_gru_100 = []
+    t_gru_200 = []
+
+    t_bert_10 = []
+    t_bert_100 = []
+    t_bert_200 = []
+
+    #LSTM
+    for com in l10:
+        com = tokenize(com)
+        t1 = time.time()
+        lstm_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_lstm_10.append(t)
+        
+    for com in l100:
+        com = tokenize(com)
+        t1 = time.time()
+        lstm_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_lstm_100.append(t)
+        
+    for com in l200:
+        com = tokenize(com)
+        t1 = time.time()
+        lstm_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_lstm_200.append(t)
+
+    #GRU
+    for com in l10:
+        com = tokenize(com)
+        t1 = time.time()
+        gru_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_gru_10.append(t)
+        
+    for com in l100:
+        com = tokenize(com)
+        t1 = time.time()
+        gru_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_gru_100.append(t)
+        
+    for com in l200:
+        com = tokenize(com)
+        t1 = time.time()
+        gru_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_gru_200.append(t)
+
+    #BERT
+    for com in l10:
+        com = tokenize(com)
+        t1 = time.time()
+        BERT_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_bert_10.append(t)
+        
+    for com in l100:
+        com = tokenize(com)
+        t1 = time.time()
+        BERT_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_bert_100.append(t)
+        
+    for com in l200:
+        com = tokenize(com)
+        t1 = time.time()
+        BERT_predict(com)
+        t2 = time.time()
+        t = t2-t1
+        t_bert_200.append(t)
+
+    return 'ok'
+
+    
+    
+        
+        
+        
+
 if __name__ == '__main__':
     
     app.run(debug=False, threaded=True)