Spaces:
Sleeping
Sleeping
| from transformers import pipeline | |
| import gradio as gr | |
| classifier = pipeline("text-classification", model="model") | |
| def detect_sentiment(text): | |
| pred = classifier(text) | |
| if len(pred) > 0: | |
| pred = pred[0] | |
| if pred['label'] == 'LABEL_1': | |
| pred['label'] = "positive" | |
| else: | |
| pred['label'] = "negative" | |
| return {pred['label'] : pred['score']} | |
| return "Internal error" | |
| review = gr.Textbox() | |
| label = gr.Label() | |
| inter = gr.Interface(fn=detect_sentiment, inputs=review, outputs=label) | |
| inter.launch(inline=False) | |