/

app.py

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