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()