Update app.py
Browse files
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 |
-
|
|
|
|
| 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 |
|