diff --git a/app/schemas/generate_image.py b/app/schemas/generate_image.py
index fee4a92..49cf9ce 100644
--- a/app/schemas/generate_image.py
+++ b/app/schemas/generate_image.py
@@ -5,6 +5,9 @@ class GenerateImageModel(BaseModel):
     tasks_id: str
     prompt: str
     image_url: str
+    mode: str
+    category: str
+    gender: str
 
 
 class GenerateSingleLogoImageModel(BaseModel):