ParshvPatel's picture
feat: HuggingFace Spaces deployment
d992912
raw
history blame contribute delete
983 Bytes
from pydantic import BaseModel
class ProductDetail(BaseModel):
sku: str
name: str
brand: str
price: float
color: str
color_family: str
category: str
gender: str
image_url: str
url: str = ""
image_urls: list[str] = []
style_tags: list[str] = []
materials: list[str] = []
sizes_available: list[str] = []
product_details: str = ""
in_stock: bool = True
class OutfitItem(BaseModel):
sku: str
name: str
brand: str
price: float
color_family: str
category: str
image_url: str
outfit_score: float
class OutfitResponse(BaseModel):
source: ProductDetail
outfit: dict[str, list[OutfitItem]]
class SimilarProductItem(BaseModel):
sku: str
name: str
brand: str
price: float
color: str
category: str
image_url: str
similarity_score: float
class SimilarResponse(BaseModel):
source: ProductDetail
results: list[SimilarProductItem]
total: int