Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,11 +14,11 @@ def predict(text):
     pred = torch.argmax(probs , dim = 1).item()
     return "Spam" if pred == 1 else "Not Spam"
 st.title("Spam Detector - BERT")
-st.write("Paste an email message to check if it is spam")
+st.write("Paste an email message to check if it's spam")
 user_input = st.text_area("Email Content", height = 200)
 if st.button("Classify"):
     if not user_input.strip():
         st.warning("Please enter some text")
 else: 
-    result.predict(user_input)
+    result = predict(user_input)
     st.sucess(f"Prediction: {result}")