mosaed-alotaibi's picture
Initial setup
e796b04 verified
Raw
History Blame Contribute Delete
848 Bytes
from transformers import pipeline
import gradio as gr
sentiment = pipeline("sentiment-analysis")
#"I really enjoyed my stay !"
#"Worst rental I ever got"
reviews = ["I really enjoyed my stay !", "Worst rental I ever got"]
results = sentiment(reviews)
print(results)
for review, result in zip(reviews, results):
print(f"review: {review}\n"
f"sentiment: {result['label']},\n"
f"score: {result['score']:.4f}\n")
def get_sentiment(text):
sentiment_result = sentiment(text)[0]
return sentiment_result['label'], sentiment_result['score']
interface = gr.Interface(
fn=get_sentiment,
inputs=gr.Textbox(lines=3, label="Enter the review"),
outputs=[
gr.Textbox(label="Sentiment"), # first return value
gr.Textbox(label="Score")
]
)
interface.launch()