Use conditional import to Pydantic v1 shim for ErrorWrapper

Signed-off-by: Mattt Zmuda <mattt@replicate.com>

python/cog/server/http.py

@@ -32,8 +32,16 @@
 from fastapi.encoders import jsonable_encoder
 from fastapi.exceptions import RequestValidationError
 from fastapi.responses import JSONResponse
-from pydantic import ValidationError
-from pydantic.error_wrappers import ErrorWrapper
+
+try:
+    from pydantic.v1 import ValidationError
+except ImportError:
+    from pydantic import ValidationError  # pylint: disable=W0404
+
+try:
+    from pydantic.v1.error_wrappers import ErrorWrapper
+except ImportError:
+    from pydantic.error_wrappers import ErrorWrapper  # pylint: disable=W0404
 
 from .. import schema
 from ..errors import PredictorNotSet