SakshamSna's picture
updated the error
4d246e5
raw
history blame contribute delete
802 Bytes
from transformers import pipeline
import gradio as gr
classifier = pipeline("sentiment-analysis")
def analyze_sentiment(text):
if not text:
return "No input", 0
try:
result = classifier(text)[0] # should return a dict
label = result.get('label', 'Unknown')
score = round(result.get('score', 0), 3)
return label, score
except Exception as e:
return f"Error: {str(e)}", 0
iface = gr.Interface(
fn=analyze_sentiment,
inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
outputs=[
gr.Textbox(label="Sentiment"),
gr.Number(label="Confidence Score")
],
title="Sentiment Analyzer",
description="Enter a sentence to get its sentiment (POSITIVE/NEGATIVE) and confidence score."
)
iface.launch()