Upload 3 files

app.py

@@ -1,9 +1,8 @@
 import streamlit as st
-import pickle
+import joblib
 
 # Load the model
-with open('best_model.pkl', 'rb') as f:
-  model = pickle.load(f)
+model = joblib.load('sms_spam_model.joblib')
 
 # Streamlit app
 st.title("SMS Spam Classifier")
@@ -11,7 +10,7 @@ message = st.text_area("Enter your message:")
 
 if st.button("Classify"):
   prediction = model.predict([message])
-  if prediction == 1:
-    st.write("This message is classified as SPAM.⚠️⚠️⚠️")
+  if prediction[0] == 1:
+    st.error("This message is classified as SPAM.")
   else:
-    st.write("This message is classified as NOT SPAM.")
+    st.success("This message is classified as NOT SPAM.")

requirements.txt

@@ -1,4 +1,3 @@
 streamlit
 nltk
-sklearn
-pickle
+sklearn

sms_spam_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:513d9048802fef7acb694e345c9ca45272331d91602850a00ef561f51d52b3da
+size 451150