Update app.py

app.py

@@ -5,6 +5,10 @@ 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)