Update app.py

app.py

@@ -3,6 +3,18 @@ import gradio as gr
 from fastai.vision.all import *
 import PIL
 import torchvision.transforms as transforms
+from albumentations import (
+    Compose,
+    OneOf,
+    ElasticTransform,
+    GridDistortion, 
+    OpticalDistortion,
+    HorizontalFlip,
+    Rotate,
+    Transpose,
+    CLAHE,
+    ShiftScaleRotate
+)
 
 class SegmentationAlbumentationsTransform(ItemTransform):
     split_idx = 0