Update app.py

app.py

@@ -6,7 +6,7 @@ from typing import Union
 
 from fastapi import FastAPI, HTTPException, Response, Request, UploadFile, File
 from fastapi.responses import StreamingResponse
-from pydantic import BaseModel, ValidationError, validator
+from pydantic import BaseModel, ValidationError, field_validator
 from transformers import (
     AutoConfig,
     AutoModelForCausalLM,
@@ -45,13 +45,15 @@ class GenerateRequest(BaseModel):
     chunk_delay: float = 0.0
     stop_sequences: list[str] = []
 
-    @validator("model_name")
+    model_config = {"protected_namespaces": ()}
+
+    @field_validator("model_name")
     def model_name_cannot_be_empty(cls, v):
         if not v:
             raise ValueError("model_name cannot be empty.")
         return v
 
-    @validator("task_type")
+    @field_validator("task_type")
     def task_type_must_be_valid(cls, v):
         valid_types = ["text-to-text", "text-to-image", "text-to-speech", "text-to-video"]
         if v not in valid_types: