Update app.py

app.py

@@ -13,6 +13,7 @@ from copy import deepcopy
 from functools import wraps
 from PIL import Image, ImageDraw
 from typing import get_type_hints, Optional
+from pydantic_core._pydantic_core import ValidationError
 # from mixpanel import Mixpanel
 
 from utils import is_not_supported_transform
@@ -136,7 +137,7 @@ def run_with_retry(compose):
                     compose.processors.pop("keypoints")
                 if "mask" in str(e):
                     kwargs.pop("mask", None)
-            except ValueError as e:
+            except (ValueError, ValidationError) as e:
                 raise gr.Error(str(e))
             except Exception as e:
                 compose.processors = processors