Spaces:
Sleeping
Sleeping
| 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() | |