Update app.py

app.py

@@ -65,7 +65,8 @@ 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.header('Predicted Class: ', prediction)
+    pred = 'Predicted Class: '+ prediction
+    st.header(pred)
     
     #st.write('Input', namestr(new_sentence, globals()),': \n', new_sentence) 
       # Remove the namestr(new_sentence, globals()) in case of an error