File size: 848 Bytes
e796b04 | 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 | 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() |