diff --git a/machine-learning/app/schemas.py b/machine-learning/app/schemas.py
index f9e64bd259..e2a027e1b8 100644
--- a/machine-learning/app/schemas.py
+++ b/machine-learning/app/schemas.py
@@ -6,6 +6,13 @@ import numpy.typing as npt
 from pydantic import BaseModel
 
 
+class StrEnum(str, Enum):
+    value: str
+
+    def __str__(self) -> str:
+        return self.value
+
+
 class TextResponse(BaseModel):
     __root__: str
 
@@ -21,12 +28,12 @@ class BoundingBox(TypedDict):
     y2: int
 
 
-class ModelType(str, Enum):
+class ModelType(StrEnum):
     CLIP = "clip"
     FACIAL_RECOGNITION = "facial-recognition"
 
 
-class ModelRuntime(str, Enum):
+class ModelRuntime(StrEnum):
     ONNX = "onnx"
     ARMNN = "armnn"