13 lines
220 B
Python
13 lines
220 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class CategoryRecognitionModel(BaseModel):
|
|
colony: str
|
|
sketch_img_url: str
|
|
|
|
|
|
class AttributeRecognitionModel(BaseModel):
|
|
category: str
|
|
colony: str
|
|
sketch_img_url: str
|