Spaces:

resolverkatla
/

Spam_Detector

Runtime error

resolverkatla commited on 20 days ago

Commit

6aadb02

1 Parent(s): cc92a16

reup

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ classifier = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-
 def spam_detector(text):
     result = classifier(text)
-    return "Spam" if result[0]['label'].lower() == "spam" else "ham"
 # Create Gradio UI
 app = gr.Interface(

 def spam_detector(text):
     result = classifier(text)
+    return "Spam" if result[0]['label'].lower() == "Spam" else "Not Spam"
 # Create Gradio UI
 app = gr.Interface(