Update app.py

app.py

@@ -1,26 +1,33 @@
 from transformers import pipeline
 import gradio as gr
 
-classifier = pipeline("text-classification", model="j-hartmann/sms-spam-detection-model")
+classifier = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
+
+label_map = {
+    "LABEL_0": "HAM (Not Spam)",
+    "LABEL_1": "SPAM"
+}
 
 def predict_spam(text):
     result = classifier(text)[0]
-    label = result['label'] 
+    label = label_map.get(result['label'], "Unknown")
     confidence = round(result['score'] * 100, 2)
-    return f" Result: {label.upper()}\n Confidence: {confidence}%"
+    return f" Prediction: {label}\n Confidence: {confidence}%"
 
+# Build Gradio UI
 gr.Interface(
     fn=predict_spam,
-    inputs=gr.Textbox(lines=4, placeholder="Enter an SMS message..."),
+    inputs=gr.Textbox(lines=4, placeholder="Enter a message..."),
     outputs="text",
-    title=" Spam Detector (Improved)",
-    description="Enter a message to check if it's spam or not using a stronger BERT model.",
+    title=" Spam Detector",
+    description="Detects SPAM vs HAM using a pretrained BERT-tiny model.",
     examples=[
-        "Congratulations! You’ve won a free cruise to the Bahamas. Reply YES to claim.",
-        "Hey, are you coming to class today?",
-        "Click here to get your free gift card now!",
-        "We need to submit the project by tonight."
+        "Win a free iPhone now!",
+        "Hey, are we still on for dinner tonight?",
+        "Your OTP is 456123.",
+        "Click here to claim your reward."
     ]
 ).launch()
 
 
+