Practica7 / app.py
nurihp's picture
Update app.py
592006c
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)