2024-09-18 00:13:01 +09:00
|
|
|
from pydantic import BaseModel
|
2024-09-30 15:24:44 +09:00
|
|
|
from typing import Literal
|
2024-09-18 00:13:01 +09:00
|
|
|
|
|
|
|
class ModelCreateRequest(BaseModel):
|
2024-09-30 15:24:44 +09:00
|
|
|
project_type: Literal["segmentation", "detection", "classification"]
|
2024-09-19 01:15:41 +09:00
|
|
|
pretrained:bool = True
|