Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| # Load model | |
| pipe = pipeline( | |
| "text-classification", | |
| model="distilbert/distilbert-base-uncased-finetuned-sst-2-english" | |
| ) | |
| # Inference function | |
| def classify_text(text): | |
| if not text.strip(): | |
| return "Please enter some text." | |
| result = pipe(text)[0] | |
| label = result["label"] | |
| score = round(result["score"], 4) | |
| return f"Sentiment: {label}\nConfidence: {score}" | |
| # Gradio UI | |
| app = gr.Interface( | |
| fn=classify_text, | |
| inputs=gr.Textbox( | |
| lines=4, | |
| placeholder="Type a sentence to analyze sentiment..." | |
| ), | |
| outputs=gr.Textbox(label="Result"), | |
| title="Sentiment Analysis App", | |
| description="Powered by DistilBERT (SST-2)\nClassifies text as POSITIVE or NEGATIVE." | |
| ) | |
| # Run app | |
| app.launch() | |