import gradio as gr
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"]}
    else:
        return {"type":"Spam", "probability":x["score"]}
        
iface = gr.Interface(
    fn=analyze_output,
    inputs=gr.Textbox(),
    outputs="text",
    live=False,
    title="Spam Classification App",
    description="Enter a text to classify it as Spam or Not Spam.",
    theme="compact"
)

iface.launch()