from fastai.vision.all import *
import gradio as gr
def is_cat(x): return x[0].isupper()
learner = load_learner('model.pkl')
categorias = ("Perro", "Gato")
def clasificar_imagen(img):
    prediccion, indice, probabilidades = learner.predict(img)
    return dict(zip(categorias, map(float,probabilidades)))
imagen = gr.inputs.Image(shape=(192,192))
etiqueta = gr.outputs.Label()
ejemplos = ['perro.jpg','gato.jpeg','gato_perro.jpeg','gato_perro_2.jpg'] 

interfaz = gr.Interface(fn=clasificar_imagen,inputs=imagen,outputs=etiqueta, examples=ejemplos)
interfaz.launch(inline=False)