13 lines
490 B
Python
13 lines
490 B
Python
|
from typing import List
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
|
||
|
|
class SketchToGarmentModel(BaseModel):
|
||
|
|
input_image_path: str = Field(..., description="输入图片路径列表")
|
||
|
|
bucket_name: str = Field(..., description="输入图片路径列表")
|
||
|
|
user_id: str = Field(..., description="用户id")
|
||
|
|
callback_url: str # 必填，客户端提供的回调地址
|
||
|
|
task_id: str = Field()
|
||
|
|
model: str = Field(default="single", description="模型类型: single 或 multi")
|