tbelway/immich-quadlet
1
0
Fork 0
forked from Cutlery/immich
Code Pull Requests Activity

fix(ml): model paths not working (#6705)

Browse Source 
* fix str enum string repr

* fixed typing
This commit is contained in:
Mert 2024-01-28 15:54:33 -05:00 committed by GitHub
parent be55396086
commit f52994e8ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
machine-learning/app/schemas.py
View File

@ -6,6 +6,13 @@ import numpy.typing as npt
from pydantic import BaseModel from pydantic import BaseModel
class StrEnum(str, Enum):
value: str
def __str__(self) -> str:
return self.value
class TextResponse(BaseModel): class TextResponse(BaseModel):
__root__: str __root__: str
@ -21,12 +28,12 @@ class BoundingBox(TypedDict):
y2: int y2: int
class ModelType(str, Enum): class ModelType(StrEnum):
CLIP = "clip" CLIP = "clip"
FACIAL_RECOGNITION = "facial-recognition" FACIAL_RECOGNITION = "facial-recognition"
class ModelRuntime(str, Enum): class ModelRuntime(StrEnum):
ONNX = "onnx" ONNX = "onnx"
ARMNN = "armnn" ARMNN = "armnn"