Update app.py

app.py

@@ -16,6 +16,9 @@ from albumentations import (
     RandomScale
 )
 
+def get_y_fn (x):
+    return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))
+
 class SegmentationAlbumentationsTransform(ItemTransform):
     split_idx = 0