diff --git a/app/schemas/generate_image.py b/app/schemas/generate_image.py
index 6fe2e70..d9ea106 100644
--- a/app/schemas/generate_image.py
+++ b/app/schemas/generate_image.py
@@ -20,8 +20,7 @@ class GenerateImageFlux2KleinModel(BaseModel):
 
 
 class GenerateImageModel(BaseModel):
-    bucket_name: str
-    object_name: str
+    tasks_id: str
     prompt: str
     image_url: str
     mode: str