HMPhuoc commited on
Commit
8e3c42c
1 Parent(s): 374606d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -42,7 +42,8 @@ def tokenizer_pad(tokenizer,comment_text,max_length=200):
42
  def LSTM_predict(x):
43
  # x = tokenizer_pad(tokenizer=tokenizer,comment_text=x)
44
 
45
- pred_proba = LSTM_model.predict(x)[0]
 
46
 
47
  pred_proba = [round(i,2) for i in pred_proba]
48
 
@@ -54,7 +55,7 @@ def GRU_predict(x):
54
  # x = tokenizer_pad(tokenizer=tokenizer,comment_text=x)
55
 
56
 
57
- pred_proba = GRU_model.predict(x)[0]
58
 
59
  pred_proba = [round(i,2) for i in pred_proba]
60
 
 
42
  def LSTM_predict(x):
43
  # x = tokenizer_pad(tokenizer=tokenizer,comment_text=x)
44
 
45
+
46
+ pred_proba = LSTM_model.predict([x])[0]
47
 
48
  pred_proba = [round(i,2) for i in pred_proba]
49
 
 
55
  # x = tokenizer_pad(tokenizer=tokenizer,comment_text=x)
56
 
57
 
58
+ pred_proba = GRU_model.predict([x])[0]
59
 
60
  pred_proba = [round(i,2) for i in pred_proba]
61