Update app.py

app.py

@@ -4,6 +4,13 @@ import gradio as gr
 # Cargamos el learner
 learn = load_learner('export_blindness.pkl')
 
+def label_func(fname):
+    labels = ['Sin RD','Leve','Moderado','Severo','RD proliferativa']
+    # blindness/train/0/e4dcca36ceb4.png
+    split = str(fname).split('/')
+    parent = int(split[2])
+    return labels[parent]
+    
 # Definimos las etiquetas de nuestro modelo
 labels = learn.dls.vocab