gad-task3 / app.py
geeaiml's picture
Create app.py
af2fd37 verified
import gradio as gr
from transformers import pipeline
# Load the sentiment analysis pipeline
sentiment_pipeline = pipeline("text-classification", model="nlptown/bert-base-multilingual-uncased-sentiment")
def analyze_sentiment(text):
if not text.strip():
return "Please enter some text to analyze."
result = sentiment_pipeline(text)[0]
return f"Predicted Sentiment: {result['label']} stars"
# Define the Gradio interface
with gr.Blocks() as demo:
gr.Markdown("# Sentiment Analysis using BERT Model")
gr.Markdown("Enter a sentence or paragraph below and click 'Analyze' to get the predicted sentiment (1 to 5 stars).")
text_input = gr.Textbox(label="Input Text", placeholder="Enter your text here...", lines=3)
analyze_button = gr.Button("Analyze Sentiment")
output_text = gr.Textbox(label="Predicted Sentiment", interactive=False)
examples = [
"I love this product! It's amazing!",
"This was the worst experience I've ever had.",
"The movie was okay, not great but not bad either.",
"Absolutely fantastic! I would recommend it to everyone."
]
gr.Examples(examples=examples, inputs=text_input)
analyze_button.click(analyze_sentiment, inputs=text_input, outputs=output_text)
# Launch the Gradio app
demo.launch()