Update app.py

app.py

@@ -5,6 +5,25 @@ from PIL import Image
 from huggingface_hub import hf_hub_download
 import torchvision.transforms as transforms
 import torch
+from albumentations import (
+    Compose,
+    HorizontalFlip,
+    Rotate,
+    GridDistortion
+)
+
+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"])
+
+
 
 # Cargar modelo desde Hugging Face Hub
 model_path = hf_hub_download(repo_id="PablitoGil14/AP-Practica3", filename="model.pkl")