import gradio as gr
import spaces
from transformers import pipeline

@spaces.GPU(duration=150)
def analyze_sentiment(text):
    classifier = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
    result = classifier(text)
    return {
        "label": result[0]["label"],
        "score": result[0]["score"]
    }

demo = gr.Interface(
    fn=analyze_sentiment,
    inputs="textbox",
    outputs="json",
    title="Sentiment Analysis",
    description="Enter text to analyze its sentiment using DistilBERT."
)

demo.queue(api_open=True)
demo.launch()