import gradio as gr
from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="distilbert-base-uncased-finetuned-sst-2-english"
)

def predict(text):
    result = classifier(text)[0]
    return f"{result['label']} ({result['score']:.2f})"

demo = gr.Interface(
    fn=predict,
    inputs="text",
    outputs="text",
    title="Simple AI Model"
)

demo.launch()