TekamBrice's picture
Update app.py
0148e34 verified
raw
history blame contribute delete
374 Bytes
import transformers
import gradio as gr
from transformers import pipeline
classifier=pipeline('sentiment-analysis',model='distilbert/distilbert-base-uncased-finetuned-sst-2-english')
def get_sentiment(text):
sentiment=classifier(text)
return sentiment[0]['label']
interface=gr.Interface(get_sentiment,title='sentiment analysis',inputs='text',outputs='text').launch()