Spaces:
Sleeping
Sleeping
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 |
|