| 123456789101112131415161718192021222324252627282930313233 |
- from enum import StrEnum
- from pydantic import BaseModel
- def to_lower_camel(string: str) -> str:
- tokens = [token.capitalize() if i > 0 else token for i, token in enumerate(string.split("_"))]
- return "".join(tokens)
- class TextModelRequest(BaseModel):
- text: str
- class TextResponse(BaseModel):
- __root__: str
- class MessageResponse(BaseModel):
- message: str
- class BoundingBox(BaseModel):
- x1: int
- y1: int
- x2: int
- y2: int
- class ModelType(StrEnum):
- IMAGE_CLASSIFICATION = "image-classification"
- CLIP = "clip"
- FACIAL_RECOGNITION = "facial-recognition"
|
