Update app.py

app.py

@@ -72,7 +72,8 @@ from fastapi.security import OAuth2PasswordRequestForm, OAuth2PasswordBearer
 from jose import JWTError, jwt
 from passlib.context import CryptContext
 from datetime import datetime, timedelta
-from typing import Optional
+from pydantic import BaseModel, field_validator, model_validator, Field, EmailStr, constr, ValidationError
+from typing import Optional, List, Union
 
 #setting up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(filename)s - %(lineno)d - %(message)s')
@@ -153,15 +154,14 @@ class GenerateRequest(BaseModel):
     mask_image: Optional[UploadFile] = None  # for image inpainting
     low_res_image: Optional[UploadFile] = None  # for image super-resolution
 
-
-    @validator("task_type")
+    @field_validator('task_type')
     def validate_task_type(cls, value):
         allowed_types = ["text", "image", "audio", "video", "classification", "translation", "question-answering", "speech-to-text", "text-to-speech", "image-segmentation", "feature-extraction", "token-classification", "fill-mask", "image-inpainting", "image-super-resolution", "object-detection", "image-captioning", "audio-transcription", "summarization"]
         if value not in allowed_types:
             raise ValueError(f"Invalid task_type. Allowed types are: {allowed_types}")
         return value
 
-    @root_validator
+    @model_validator(mode='after')
     def check_input(cls, values):
         task_type = values.get("task_type")
         if task_type == "text" and values.get("input_text") is None:
@@ -182,8 +182,6 @@ class GenerateRequest(BaseModel):
             raise ValueError("low_res_image is required for image super-resolution.")
         return values
 
-
-
 class S3ModelLoader:
     def __init__(self, bucket_name, aws_access_key_id, aws_secret_access_key, aws_region):
         self.bucket_name = bucket_name
@@ -688,4 +686,4 @@ if __name__ == "__main__":
 
     create_db_and_table() # Initialize database on startup
 
-    uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=True)  # replace main with your filename
+    uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=True)