added new

app.py

@@ -1,12 +1,17 @@
 import gradio as gr
 from transformers import pipeline
+import pandas as pd
 
-# Load the spam classifier model
+# Load dataset
+DATASET_PATH = "spam.csv"
+df = pd.read_csv(DATASET_PATH)
+
+# Load a spam classification model
 classifier = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
 
 def spam_detector(text):
     result = classifier(text)
-    return "Spam" if result[0]['label'] == 'spam' else "Not Spam"
+    return "Spam" if result[0]['label'].lower() == "spam" else "ham"
 
 # Create Gradio UI
 app = gr.Interface(
@@ -19,4 +24,7 @@ app = gr.Interface(
 
 # Run the app
 if __name__ == "__main__":
+    print("Loaded dataset preview:")
+    print(df.head())
     app.launch()
+

requirements.txt

@@ -3,4 +3,5 @@ pipeline
 gradio
 tensorflow
 torch
-tf-keras
+tf-keras
+pandas