| from transformers import pipeline |
| import gradio as gr |
|
|
| sentiment = pipeline("sentiment-analysis") |
|
|
| |
| |
|
|
| 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"), |
| gr.Textbox(label="Score") |
| ] |
| ) |
|
|
|
|
| interface.launch() |