k-code's picture
push to hugging face
24b3cda
metadata
title: SMS Spam Classifier
emoji: 📱
colorFrom: blue
colorTo: red
sdk: gradio
sdk_version: 4.0.0
app_file: app.py
pinned: false

SMS Spam Classifier

This application uses a bidirectional LSTM model to classify SMS messages as either spam or legitimate (ham). Simply enter your text message, and the model will predict whether it's spam or not, along with a confidence score.

Usage

  1. Enter your text message in the input box
  2. Click submit
  3. The model will return its prediction (spam/ham) and confidence level

Model

The classifier uses a bidirectional LSTM architecture with:

  • Word embeddings
  • 2 LSTM layers
  • Dropout for regularization
  • Dense layers with ReLU activation