import streamlit as st
import pickle

# Load the model
with open('best_model.pkl', 'rb') as f:
  model = pickle.load(f)

# Streamlit app
st.title("SMS Spam Classifier")
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.⚠️⚠️⚠️")
  else:
    st.write("This message is classified as NOT SPAM.")