Update app.py

app.py

@@ -2,9 +2,13 @@ import gradio as gr
 from transformers import pipeline
 
 def analyze_output(input_text):
-    pipe = pipeline("text-classification", model="ZachBeesley/Spam-Detector")
-    result = pipe(input_text)[0]
-    return str(result)
+    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"]}
+    else:
+        return {"type":"Spam", "probability":x["score"]}
+        
 iface = gr.Interface(
     fn=analyze_output,
     inputs=gr.Textbox(),