Files
sora_python/app/schemas/outfit_matcher.py
2024-03-18 14:17:43 +08:00

16 lines
274 B
Python

from pydantic import BaseModel
class OMDataItem(BaseModel):
item_name: str
semantic_category: str
image_path: str
class OutfitMatcher(BaseModel):
topk: int
max_outfits: int
is_best: bool
query: list[OMDataItem]
database: list[OMDataItem]