Spaces:
Running
Running
| import gradio as gr | |
| from transformers import pipeline | |
| # Load sentiment analysis pipeline | |
| classifier = pipeline( | |
| "sentiment-analysis", | |
| model="distilbert-base-uncased-finetuned-sst-2-english", | |
| return_all_scores=True | |
| ) | |
| def analyze_sentiment(text): | |
| """Analyze sentiment and return percentages.""" | |
| if not text.strip(): | |
| return {"Positive": 0, "Negative": 0, "Neutral": 0}, "" | |
| results = classifier(text)[0] | |
| # Convert to percentage format | |
| sentiment_map = {} | |
| for r in results: | |
| label = r['label'] | |
| score = r['score'] * 100 | |
| if label == 'POSITIVE': | |
| sentiment_map['Positive'] = round(score, 2) | |
| elif label == 'NEGATIVE': | |
| sentiment_map['Negative'] = round(score, 2) | |
| # Add neutral if needed (for models that only have pos/neg) | |
| if 'Neutral' not in sentiment_map: | |
| sentiment_map['Neutral'] = round(100 - sentiment_map.get('Positive', 0) - sentiment_map.get('Negative', 0), 2) | |
| # Create summary text | |
| dominant = max(sentiment_map, key=sentiment_map.get) | |
| summary = f"**Dominant: {dominant} ({sentiment_map[dominant]}%)**" | |
| return sentiment_map, summary | |
| # Create Gradio interface | |
| with gr.Blocks(title="Simple Sentiment Analyzer") as demo: | |
| gr.Markdown("# 🎭 Simple Sentiment Analyzer") | |
| gr.Markdown("Enter a sentence to see sentiment breakdown with percentages.") | |
| with gr.Row(): | |
| with gr.Column(): | |
| text_input = gr.Textbox( | |
| label="Input Text", | |
| placeholder="Type your sentence here...", | |
| lines=4 | |
| ) | |
| analyze_btn = gr.Button("Analyze", variant="primary") | |
| with gr.Column(): | |
| output_json = gr.Label(label="Sentiment Scores (%)", num_top_classes=3) | |
| summary_md = gr.Markdown() | |
| analyze_btn.click( | |
| fn=analyze_sentiment, | |
| inputs=text_input, | |
| outputs=[output_json, summary_md] | |
| ) | |
| # Example inputs | |
| gr.Examples( | |
| examples=[ | |
| "I absolutely love this product! It's amazing!", | |
| "This is terrible. I hate it.", | |
| "The weather is okay today.", | |
| "The movie was okay, nothing special but not bad either.", | |
| "Best day ever! Everything went perfectly!" | |
| ], | |
| inputs=text_input | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() |