mirror of
https://github.com/immich-app/immich.git
synced 2024-12-27 10:58:13 +02:00
34201be74c
* using pydantic BaseSetting * ML API takes image file as input * keeping image in memory * reducing duplicate code * using bytes instead of UploadFile & other small code improvements * removed form-multipart, using HTTP body * format code --------- Co-authored-by: Alex Tran <alex.tran1502@gmail.com>
57 lines
934 B
Python
57 lines
934 B
Python
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 TagResponse(BaseModel):
|
|
__root__: list[str]
|
|
|
|
|
|
class Embedding(BaseModel):
|
|
__root__: list[float]
|
|
|
|
|
|
class EmbeddingResponse(BaseModel):
|
|
__root__: Embedding
|
|
|
|
|
|
class BoundingBox(BaseModel):
|
|
x1: int
|
|
y1: int
|
|
x2: int
|
|
y2: int
|
|
|
|
|
|
class Face(BaseModel):
|
|
image_width: int
|
|
image_height: int
|
|
bounding_box: BoundingBox
|
|
score: float
|
|
embedding: Embedding
|
|
|
|
class Config:
|
|
alias_generator = to_lower_camel
|
|
allow_population_by_field_name = True
|
|
|
|
|
|
class FaceResponse(BaseModel):
|
|
__root__: list[Face]
|