From aec03be2c98ecc1f90714e94424f505699ecbb39 Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Wed, 12 Jun 2024 15:32:25 +0800
Subject: [PATCH] =?UTF-8?q?feat=20generate=20product=20image=20=E9=83=A8?=
 =?UTF-8?q?=E7=BD=B2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/schemas/generate_image.py | 3 +++
 1 file changed, 3 insertions(+)

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):