feat generate 升级 attribute retrieve 迁移

app/schemas/attribute_retrieve.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel
+
+
+class CategoryRecognitionModel(BaseModel):
+    colony: str
+    sketch_img_url: str
+
+
+class AttributeRecognitionModel(BaseModel):
+    category: str
+    colony: str
+    sketch_img_url: str

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