fix(ml): pydantic dep causes starting up issue (#11773)

* fix(ml): pydantic dep causes starting up issue

* revert import

machine-learning/app/config.py

@@ -6,7 +6,7 @@ from pathlib import Path
 from socket import socket
 
 from gunicorn.arbiter import Arbiter
-from pydantic.v1 import BaseModel, BaseSettings
+from pydantic import BaseModel, BaseSettings
 from rich.console import Console
 from rich.logging import RichHandler
 from uvicorn import Server

machine-learning/app/main.py

@@ -15,7 +15,7 @@ from fastapi import Depends, FastAPI, File, Form, HTTPException
 from fastapi.responses import ORJSONResponse
 from onnxruntime.capi.onnxruntime_pybind11_state import InvalidProtobuf, NoSuchFile
 from PIL.Image import Image
-from pydantic.v1 import ValidationError
+from pydantic import ValidationError
 from starlette.formparsers import MultiPartParser
 
 from app.models import get_model_deps

machine-learning/app/schemas.py

@@ -3,7 +3,7 @@ from typing import Any, Literal, Protocol, TypedDict, TypeGuard, TypeVar
 
 import numpy as np
 import numpy.typing as npt
-from pydantic.v1 import BaseModel
+from pydantic import BaseModel
 
 
 class StrEnum(str, Enum):