Space_Test / app.py
AbdulrahmanJ's picture
Create app.py
8909fa5 verified
raw
history blame contribute delete
790 Bytes
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()