Update app.py

app.py

@@ -70,11 +70,10 @@ def predict(new_sentence):
       output = model(test_ids.to(device), token_type_ids = None, attention_mask = test_attention_mask.to(device))
 
     prediction = 'Spam' if np.argmax(output.logits.cpu().numpy()).flatten().item() == 1 else 'Normal'
-
+    st.write('Predicted Class: ', prediction)
     
-    st.write('Input', namestr(new_sentence, globals()),': \n', new_sentence) 
+    #st.write('Input', namestr(new_sentence, globals()),': \n', new_sentence) 
       # Remove the namestr(new_sentence, globals()) in case of an error      
-    st.write('Predicted Class: ', prediction,'\n----------------------------------\n')
 if st.button('Analyze'):
     predict(text)