import pandas as pd
from fastai.text.all import *
from datasets import load_dataset
from transformers import pipeline
import gradio as gr

# Modelo de clasificación usando LSTM
#learner = load_learner('modelLSTM.pkl')

# Modelo de clasificación de texto usando modelos de lenguaje
#learner = load_learner('modelML.pkl')

# Modelo de clasificación basados en mecanismos de atención
classifier = pipeline('text-classification', model='edgilr/clasificador-rotten-tomatoes')

def predict(txt):
    # Modelo de clasificación usando LSTM o modelo de clasificación de texto usando modelos de lenguaje
    #pred,pred_idx,probs = learner.predict(txt)
    #return pred
    # Modelo de clasificación basados en mecanismos de atención
    return classifier(txt)[0]['label']
    
gr.Interface(fn=predict, inputs="text", outputs="text",
             examples=['the story gives ample opportunity for large-scale action and suspense , which director shekhar kapur supplies with tremendous skill .',
                       'the thing looks like a made-for-home-video quickie .']).launch(share=True)