feat(ml): conditionally download .armnn models (#6650)

2024-01-28 10:31:59 -05:00
parent fa0913120d
commit a84b6f5fb1
5 changed files with 127 additions and 38 deletions
--- a/machine-learning/app/schemas.py
+++ b/machine-learning/app/schemas.py
@ -26,6 +26,11 @@ class ModelType(str, Enum):
    FACIAL_RECOGNITION = "facial-recognition"


+class ModelRuntime(str, Enum):
+    ONNX = "onnx"
+    ARMNN = "armnn"
+
+
 class HasProfiling(Protocol):
    profiling: dict[str, float]