Update app.py

app.py

@@ -3,11 +3,11 @@ from transformers import pipeline
 
 def analyze_output(input_text):
     pipe = pipeline("text-classification", model="Titeiiko/OTIS-Official-Spam-Model")
-    x = pipe(input)[0]
-    if x["label"] == "LABEL_0":
-        return {"type":"Not Spam", "probability":x["score"]}
+    result = pipe(input_text)[0]
+    if result["label"] == "LABEL_0":
+        return f"Not Spam: {result['score']:.4f}"
     else:
-        return {"type":"Spam", "probability":x["score"]}
+        return f"Spam: {result['score']:.4f}"
         
 iface = gr.Interface(
     fn=analyze_output,