Update app.py

app.py

@@ -52,6 +52,12 @@ class TargetMaskConvertTransform(ItemTransform):
 
 learn = load_learner(model_path)
 
+
+def get_y_fn(x):
+    return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))
+
+
+
 def segmentar(img: Image.Image):
     img = img.resize((640, 480))
     x = transforms.Compose([