import pandas as pd
from fastai.text.all import *
from datasets import load_dataset

import gradio as gr


# Cargamos el learner
learn = load_learner('export.pkl')

# Definimos las etiquetas de nuestro modelo
labels = learn.dls.vocab


# Definimos una función que se encarga de llevar a cabo las predicciones
def predict(text):
    #img = PILImage.create(img)
    pred,pred_idx,probs = learn.predict(text)
    return pred
    
# Creamos la interfaz y la lanzamos. 
gr.Interface(fn=predict, inputs="text", outputs="text",examples=['Jeg vil også benytte min tilstedeværelse her i Europa-Parlamentet til at redegøre for, hvad jeg allerede nævnte under høringen i Økonomi- og Valutaudvalget i mandags.','Inddragelse af støtten må imidlertid ikke overvejes.']).launch(share=False)