Update app.py
Browse files
app.py
CHANGED
@@ -71,9 +71,9 @@ def predict(new_sentence):
|
|
71 |
prediction = 'Spam' if np.argmax(output.logits.cpu().numpy()).flatten().item() == 1 else 'Normal'
|
72 |
|
73 |
|
74 |
-
st.
|
75 |
# Remove the namestr(new_sentence, globals()) in case of an error
|
76 |
-
st.
|
77 |
|
78 |
predict(text)
|
79 |
|
|
|
71 |
prediction = 'Spam' if np.argmax(output.logits.cpu().numpy()).flatten().item() == 1 else 'Normal'
|
72 |
|
73 |
|
74 |
+
st.write('Input', namestr(new_sentence, globals()),': \n', new_sentence)
|
75 |
# Remove the namestr(new_sentence, globals()) in case of an error
|
76 |
+
st.write('Predicted Class: ', prediction,'\n----------------------------------\n')
|
77 |
|
78 |
predict(text)
|
79 |
|