Spaces:
Runtime error
Runtime error
File size: 590 Bytes
769d4a6 5685f42 769d4a6 dd0e634 3a96c51 8228a46 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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() |