Upload schemas.py

schemas.py

@@ -30,14 +30,18 @@ class CustomModeGenerateParam(BaseModel):
     )
     continue_clip_id: Optional[str] = None
 
-    negative_tags: str = ""
+    generation_type: str = "TEXT"
+    negative_tags:list = []
+    infill_start_s:Optional[int] = None
+    infill_end_s:Optional[int] = None
+    task:Optional[str] = ""
 
 
 class DescriptionModeGenerateParam(BaseModel):
     """Generate with Song Description"""
 
-    gpt_description_prompt: str
-    make_instrumental: bool = False
+    gpt_description_prompt:str = Field( max_length=200)
+    make_instrumental: bool = False # True纯音乐，False有歌词
     mv: str = Field(
         default='chirp-v3-0',
         description="model version, default: chirp-v3-0",
@@ -50,4 +54,4 @@ class DescriptionModeGenerateParam(BaseModel):
     )
 
     generation_type: str = "TEXT"
-    user_uploaded_images_b64: str = ""
+    user_uploaded_images_b64:List[str] = []