from transformers import BertTokenizer, BertModel
from transformers import pipeline
import gradio as grad

model_name = "MiVaCod/rotten"
text2text_tkn= BertTokenizer.from_pretrained(model_name)
mdl = BertModel.from_pretrained(model_name)

def text2text_paraphrase(sentence1):
    classifier = pipeline('text-classification', model='MiVaCod/rotten')
    res = classifier(sentence1)

    clase = res[0]
    if res=='LABEL_0':
        res = 'BAD!'
    else:
        res = 'GOOD!'
    
    return res

sent1=grad.Textbox(lines=1, label="Review", placeholder="Introduce la review de una película.")
grad.Interface(text2text_paraphrase, inputs=sent1, outputs="text").launch()