Spaces:
Runtime error
Runtime error
| from transformers import pipeline | |
| import gradio as gr | |
| sentiment_analysis = pipeline("sentiment-analysis",model="siebert/sentiment-roberta-large-english") | |
| # print(sentiment_analysis("I love this!")) | |
| def model(a): | |
| x=sentiment_analysis(a)[0] | |
| if x['label']=='POSITIVE': | |
| d={x['label']:x['score'],'NEGATIVE':1-x['score']} | |
| else: | |
| d={'POSITIVE':1-x['score'],x['label']:x['score']} | |
| return d | |
| demo = gr.Interface(fn=model, inputs="text", outputs="label") | |
| demo.launch() | |
| # import gradio as gr | |
| # gr.Interface.load("models/siebert/sentiment-roberta-large-english").launch() |