feat generate 升级 attribute retrieve 迁移

2024-04-16 15:51:03 +08:00
parent 6b9fd16244
commit 835943b568
64 changed files with 1397 additions and 201 deletions
--- a/app/schemas/generate_image.py
+++ b/app/schemas/generate_image.py
@@ -2,11 +2,8 @@ from pydantic import BaseModel


 class GenerateImageModel(BaseModel):
-    category: str
-    content: str
-    gender: str
-    image_url: str
-    mode: int
    tasks_id: str
-    user_id: int
-    version: str
+    prompt: str
+    image_url: str
+    mode: str
+    category: str