Update app.py

app.py

@@ -4,6 +4,17 @@ from fastai.vision.all import *
 import PIL
 import torchvision.transforms as transforms
 
+class SegmentationAlbumentationsTransform(ItemTransform):
+    split_idx = 0
+    
+    def __init__(self, aug): 
+        self.aug = aug
+        
+    def encodes(self, x):
+        img,mask = x
+        aug = self.aug(image=np.array(img), mask=np.array(mask))
+        return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
+
 class TargetMaskConvertTransform(ItemTransform):
     def __init__(self): 
         pass