File size: 1,346 Bytes
07e0380
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import gradio as gr
from transformers import pipeline

def analyze_sentiment(text):
    sentiment_analyzer = pipeline(
        "sentiment-analysis",
        model="nlptown/bert-base-multilingual-uncased-sentiment"
    )
    result = sentiment_analyzer(text, return_all_scores=True)
    score = int(result[0]['score'] * 5)  # Convert probability to 5-star scale
    stars = "⭐" * score
    
    return stars

with gr.Blocks() as demo:
    gr.Markdown("## Sentiment Analysis Demo")
    with gr.Row():
        input_text = gr.Textbox(
            label="Enter your text here",
            placeholder="Type or paste your text...",
            lines=3,
            examples=[
                "I love this product! It's amazing!",
                "This was the worst experience I've ever had.",
                "The movie was okay, not great but not bad either.",
                "Absolutely fantastic! I would recommend it to everyone."
            ]
        )
    with gr.Row():
        analyze_button = gr.Button("Analyze Sentiment", variant="primary")
    with gr.Row():
        output_text = gr.Textbox(
            label="Sentiment (Stars)",
            lines=1
        )
        
    analyze_button.click(
        fn=analyze_sentiment,
        inputs=input_text,
        outputs=output_text
    )

if __name__ == "__main__":
    demo.launch()