| import gradio as gr | |
| from transformers import pipeline | |
| classifiers = pipeline( | |
| "text-classification", | |
| model="distilbert/distilbert-base-uncased-finetuned-sst-2-english", | |
| ) | |
| def predict_my_sentiment(name): | |
| prediction = classifiers(name)[0] | |
| label = prediction['label'] | |
| score = prediction['score'] | |
| return label,score | |
| demo = gr.Interface( | |
| fn=predict_my_sentiment, | |
| inputs=gr.Textbox(label="Type Your Current mindset !!!"), | |
| outputs=[ | |
| gr.Textbox(label="Sentiment Label"), | |
| gr.Number(label="Confidence Score") | |
| ] | |
| ) | |
| demo.launch() | |