Spaces:
Sleeping
Sleeping
from transformers import pipeline | |
import gradio as gr | |
# Load the sentiment analysis model | |
sentiment_analysis = pipeline("sentiment-analysis") | |
# Define the prediction function | |
def predict_sentiment(text): | |
result = sentiment_analysis(text)[0] | |
label = result['label'] | |
confidence = round(result['score'], 4) | |
return f"Sentiment: {label}, Confidence: {confidence}" | |
# Create a Gradio interface | |
interface = gr.Interface(fn=predict_sentiment, | |
inputs=gr.inputs.Textbox(lines=2, placeholder="Type your text here..."), | |
outputs="text", | |
title="Text Sentiment Analysis", | |
description="This tool predicts the sentiment of the entered text. Sentiment can be positive, negative, or neutral.") | |
# Launch the application | |
interface.launch() | |