File size: 790 Bytes
8909fa5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import gradio as gr
from transformers import pipeline

sentiment_pipeline = pipeline("sentiment-analysis")

reviews = ["I really enjoyed my stay!", "Worst rental I ever got"]
results = sentiment_pipeline(reviews)

for review, result in zip(reviews, results):
    print(f"Review: {review}")
    print(f"Sentiment: {result['label']}, Confidence: {result['score']:.4f}\n")

def get_sentiment(text):
    result = sentiment_pipeline(text)[0]  # Extract first result
    sentiment = result["label"]
    confidence = result["score"]
    return sentiment, confidence

interface = gr.Interface(fn = get_sentiment, inputs=gr.Textbox(label = "Enter the review :"), outputs = [gr.Textbox(label = "Sentiment :"), gr.Textbox(label = "Score :")], title = "Sentiment analysis protoype")
interface.launch()