Update app.py

app.py

@@ -2,6 +2,9 @@ from huggingface_hub import from_pretrained_fastai
 import gradio as gr
 from fastai.vision.all import *
 
+def get_y_fn (x):
+    return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))
+
 from albumentations import (
     Compose,
     OneOf,