Update app.py

app.py

@@ -21,6 +21,20 @@ class TargetMaskConvertTransform(ItemTransform):
         mask = PILMask.create(mask)
         return img, mask
 
+# Define la clase SegmentationAlbumentationsTransform
+class SegmentationAlbumentationsTransform(ItemTransform):
+    def __init__(self):
+        pass
+
+    def encodes(self, x):
+        img, mask = x
+        # Aquí deberías definir tu transformación de Albumentations
+        # Por ejemplo, podrías tener algo como:
+        # transformed = my_albumentations_function(image=img, mask=mask)
+        # return transformed['image'], transformed['mask']
+        return img, mask
+
+      
 # Carga el modelo después de definir la clase
 repo_id = "LuisCe/Practica03"
 learner = from_pretrained_fastai(repo_id)