kingslin's picture
Update app.py
a26e419 verified
raw
history blame contribute delete
561 Bytes
import gradio as gr
from transformers import pipeline
# Load sentiment analysis pipeline
classifier = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
# Define prediction function
def predict(text):
result = classifier(text)[0]
return result["label"], round(result["score"], 3)
# Create Gradio interface
app = gr.Interface(
fn=predict,
inputs=gr.Textbox(label="Enter text"),
outputs=[gr.Text(label="Sentiment"), gr.Number(label="Confidence Score")]
)
# 🚀 This is required in Spaces
app.launch()