Madiharehan's picture
Create app.py
01c3c58 verified
raw
history blame
1.09 kB
import gradio as gr
from transformers import pipeline
# Load the pre-trained model (cached for performance)
def load_model():
return pipeline('sentiment-analysis', model='cardiffnlp/twitter-roberta-base-sentiment')
sentiment_model = load_model()
# Define the function to analyze sentiment
def analyze_sentiment(user_input):
result = sentiment_model(user_input)[0]
sentiment = result['label']
if sentiment in ['NEGATIVE', 'NEUTRAL']:
return "Stay positive! 🌟 You can handle anything that comes your way."
return "You're on the right track! Keep shining! 🌞"
# Gradio UI
def chatbot_ui():
# Define the interface
interface = gr.Interface(
fn=analyze_sentiment,
inputs=gr.Textbox(label="Enter your text here:"),
outputs=gr.Textbox(label="Motivational Message"),
title="Student Sentiment Analysis Chatbot",
description="This chatbot detects your mood and provides positive or motivational messages."
)
return interface
# Launch the interface
if __name__ == "__main__":
chatbot_ui().launch()