Create app.py

app.py

@@ -0,0 +1,28 @@
+
+
+import streamlit as st
+import pickle
+
+# loading the trained model
+model = pickle.load(open('model.pkl', 'rb'))
+
+# create title
+st.title('Predicting if message is spam or not')
+
+message = st.text_input('Enter a message')
+
+submit = st.button('Predict')
+
+if submit:
+    prediction = model.predict([message])
+
+    # print(prediction)
+    # st.write(prediction)
+    
+    if prediction[0] == 'spam':
+        st.warning('This message is spam')
+    else:
+        st.success('This message is Legit (HAM)')
+
+
+st.balloons()