Update app.py

app.py

@@ -248,10 +248,12 @@ def update_augmented_images(image, code):
     try:
         augmentation = eval(code)
     except ValidationError as e:
+        raise gr.Error(str(e))
+    except Exception as e:
         logger.info(code)
         logger.error(e)
-        raise gr.Error(str(e))
-
+        raise e
+        
     track_event("transform_applied", properties={"transform_name": augmentation.__class__.__name__, "code": code})
 
     compose = A.Compose(