diff --git a/app/schemas/generate_image.py b/app/schemas/generate_image.py
index 3dd7cf8..11e295f 100644
--- a/app/schemas/generate_image.py
+++ b/app/schemas/generate_image.py
@@ -8,6 +8,7 @@ class GenerateImageModel(BaseModel):
     mode: str
     category: str
     gender: str
+    version: str
 
 
 class GenerateSingleLogoImageModel(BaseModel):