Update server.py

server.py

@@ -5,7 +5,9 @@ from paddleocr import PaddleOCR
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from passporteye import read_mrz
-from pydantic.v1 import BaseModel, Field
+from pydantic.v1 import BaseModel as v1BaseModel
+from pydantic.v1 import Field
+from pydantic import BaseModel
 from typing import Any, Optional, Dict, List
 from huggingface_hub import InferenceClient
 from langchain.llms.base import LLM
@@ -18,7 +20,7 @@ model_name = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 hf_token = HF_token
 kwargs = {"max_new_tokens":500, "temperature":0.1, "top_p":0.95, "repetition_penalty":1.0, "do_sample":True}
 
-class KwArgsModel(BaseModel):
+class KwArgsModel(v1BaseModel):
     kwargs: Dict[str, Any] = Field(default_factory=dict)
 
 class CustomInferenceClient(LLM, KwArgsModel):