test_01 / app.py
ereeyees's picture
app
6945fce verified
raw
history blame contribute delete
687 Bytes
import gradio as gr
from transformers import pipeline
# Load the sentiment analysis pipeline from Hugging Face
sentiment_pipeline = pipeline("sentiment-analysis")
def analyze_sentiment(text):
result = sentiment_pipeline(text)[0]
sentiment = result['label']
confidence = result['score']
return f"Sentiment: {sentiment}, Confidence: {confidence:.2f}"
# Create the Gradio interface
iface = gr.Interface(
fn=analyze_sentiment,
inputs=gr.inputs.Textbox(lines=2, placeholder="Enter text here..."),
outputs="text",
title="Sentiment Analysis",
description="Enter text and click the button to detect the sentiment."
)
# Launch the interface
iface.launch()