keerthii05's picture
Create app.py
271eefc verified
Raw History Blame Contribute Delete
534 Bytes
import gradio as gr
from transformers import pipeline
# Load pre-trained sentiment-analysis pipeline
model = pipeline("sentiment-analysis", model="distilbert-base-uncased")
# Function to classify sentiment
def analyze_sentiment(feedback_text):
result = model(feedback_text)
return result[0]['label'], result[0]['score'] # returns label and confidence score
# Gradio interface
iface = gr.Interface(fn=analyze_sentiment,
inputs="text",
outputs=["text", "number"])
iface.launch()