Update app.py

app.py

@@ -245,7 +245,11 @@ def update_augmented_images(image, code):
     if "=," in code:
         raise gr.Error("You have to fill in some parameters to apply transform!")
 
-    augmentation = eval(code)
+    try:
+        augmentation = eval(code)
+    except ValidationError as e:
+        gr.Error(str(e))
+
     track_event("transform_applied", properties={"transform_name": augmentation.__class__.__name__, "code": code})
 
     compose = A.Compose(