app.py

from fastai.vision.all import *
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(img):
    img = PILImage.create(img)
    pred = learn.predict(img)[0]
    return pred
 
#
    
# Creamos la interfaz y la lanzamos. 
# gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(128, 128)), outputs="text",examples=['anonymized_1284335.jpg','anonymized_1326804.jpg']).launch(share=False)
image = gr.Image(shape=(128, 128))
label = gr.Label()

intf = gr.Interface(fn=predict, inputs=image, outputs=label, examples=['anonymized_1284335.jpg','anonymized_1326804.jpg'])
intf.launch(inline=False)