From 19ee48f6f0291bbad4cc7230a6e6ba5f06d8d380 Mon Sep 17 00:00:00 2001 From: yoni13 Date: Sun, 12 Jan 2025 01:09:36 +0800 Subject: [PATCH] fix path --- machine-learning/app/sessions/rknn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/machine-learning/app/sessions/rknn.py b/machine-learning/app/sessions/rknn.py index 0606cef4c2..9348b3770f 100644 --- a/machine-learning/app/sessions/rknn.py +++ b/machine-learning/app/sessions/rknn.py @@ -24,9 +24,9 @@ class RknnSession: self.model_path = Path(model_path) self.ort_model_path = str(self.model_path).replace(".rknn", ".onnx") - if 'textual' in self.model_path.name: + if 'textual' in str(self.model_path): self.tpe = settings.rknn_textual_threads - elif 'visual' in self.model_path.name: + elif 'visual' in str(self.model_path): self.tpe = settings.rknn_visual_threads else: self.tpe = settings.rknn_facial_detection_threads