File size: 326 Bytes
00e79b6
 
 
 
 
 
 
 
 
 
 
356df8d
1
2
3
4
5
6
7
8
9
10
11
12
from transformers import pipeline
import gradio as gr

classifier = pipeline("sentiment-analysis")

def predict(prompt):
    result = classifier(prompt)
    label = result[0]["label"]
    score = result[0]["score"]
    return label + " " + str(score)

gr.Interface(fn=predict, inputs="text", outputs="text").launch(share=True)