Update app.py

app.py

@@ -12,18 +12,7 @@ from pathlib import Path
 import random
 import PIL
 import torchvision.transforms as transforms
-from albumentations import (
-    Compose,
-    OneOf,
-    ElasticTransform,
-    GridDistortion, 
-    OpticalDistortion,
-    HorizontalFlip,
-    Rotate,
-    Transpose,
-    CLAHE,
-    ShiftScaleRotate
-)
+from albumentations import *
 
 def get_y_fn (x):
     return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))