Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| import joblib | |
| import os | |
| # ====================== | |
| # LOAD MODEL | |
| # ====================== | |
| BASE_DIR = os.path.dirname(os.path.abspath(__file__)) | |
| model = joblib.load(os.path.join(BASE_DIR, "model.pkl")) | |
| vectorizer = joblib.load(os.path.join(BASE_DIR, "vectorizer.pkl")) | |
| # ====================== | |
| # PAGE | |
| # ====================== | |
| st.set_page_config(page_title="SMS Spam Detection", page_icon="📩") | |
| st.title("📩 SMS Spam Detection") | |
| st.write("Classify SMS messages as Spam or Ham") | |
| # ====================== | |
| # INPUT | |
| # ====================== | |
| message = st.text_area("Enter your message") | |
| # ====================== | |
| # PREDICT | |
| # ====================== | |
| if st.button("Predict"): | |
| message_vec = vectorizer.transform([message]) | |
| prediction = model.predict(message_vec)[0] | |
| proba = model.predict_proba(message_vec) | |
| confidence = proba.max() | |
| if prediction == "spam": | |
| st.error(f"🚫 Spam ({confidence:.2%})") | |
| else: | |
| st.success(f"✅ Ham ({confidence:.2%})") |