Update app.py

app.py

@@ -5,6 +5,11 @@ from fastai.vision.all import *
 from fastai.learner import load_learner
 from PIL import Image
 
+# Define la función get_y_fn si se utiliza en tu modelo
+
+def get_y_fn (x):
+    return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))
+
 # Define la clase TargetMaskConvertTransform
 class TargetMaskConvertTransform(ItemTransform):
     def __init__(self):
@@ -20,7 +25,7 @@ class TargetMaskConvertTransform(ItemTransform):
         mask[mask == 25] = 4
         mask = PILMask.create(mask)
         return img, mask
-
+        
 # Define la clase SegmentationAlbumentationsTransform
 class SegmentationAlbumentationsTransform(ItemTransform):
     def __init__(self):