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)