Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from fastai.vision.all import *
 from huggingface_hub import from_pretrained_fastai
 
+def label_func(fn): return path/'masks1b-binary'/f'{fn.stem}.png'
+
 repo_id = "hugginglearners/kvasir-seg"
 learn = from_pretrained_fastai(repo_id)
 #labels = learn.dls.vocab