RohitCSharp's picture
Update app.py
97a62f2 verified
raw
history blame contribute delete
332 Bytes
import gradio as gr
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="bert-base-uncased")
def classify_sentiment(text):
result = classifier(text)[0]
return f"{result['label']} ({round(result['score'] * 100, 2)}%)"
gr.Interface(fn=classify_sentiment, inputs="text", outputs="text").launch()