SpamShield AI is a modern SMS Spam Detection system built using Machine Learning, NLP, and Streamlit.
It classifies messages as Spam or Ham (Not Spam) using TF‑IDF + Logistic Regression and includes a clean UI.
- ✔ Detects SMS spam with high accuracy
- ✔ Clean, fast Streamlit UI
- ✔ Real-time predictions
- ✔ NLP preprocessing & TF‑IDF vectorization
- ✔ Fully open-source and extensible
| Layer | Technology | Purpose |
|---|---|---|
| Machine Learning | Scikit‑Learn | Spam classification |
| NLP | TF‑IDF Vectorizer | Text representation |
| Frontend | Streamlit | UI for predictions |
| Backend | Python | ML inference |
| Storage | joblib | Model persistence |
SpamShield-AI/
│── data/
│ └── spam.csv
│── models/
│ └── spam_model.pkl
│── app.py
│── train.py
│── requirements.txt
└── README.md
git clone https://github.com/ramarav/SpamShield-AI
cd SpamShield-AI
pip install -r requirements.txtpython train.pyThis creates:
models/spam_model.pkl
models/tfidf.pkl
streamlit run app.py- Input: "Congratulations! You've won a prize. Click claim now!"
- Output: Spam 🚫
- Add LLM‑based explanations
- Deploy as a web service
- Add dashboard analytics
Pull requests are welcome.
MIT License.