Update app.py

app.py

@@ -71,9 +71,9 @@ def predict(new_sentence):
     prediction = 'Spam' if np.argmax(output.logits.cpu().numpy()).flatten().item() == 1 else 'Normal'
 
     
-    st.header('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.header('Predicted Class: ', prediction,'\n----------------------------------\n')
+    st.write('Predicted Class: ', prediction,'\n----------------------------------\n')
     
 predict(text)