fix install error

coder168 · Jul 13, 2023 · 81923e3 · 81923e3
1 parent 37db1f6
commit 81923e3
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/scripts/inferencers/insightface/installer.py b/scripts/inferencers/insightface/installer.py
@@ -8,4 +8,13 @@ def name(self) -> str:
         return "InsightFace"
 
     def requirements(self) -> List[str]:
-        return ["insightface"]
+        return ['"insightface>=0.7.3"', "onnxruntime"]
+
+    def install(self) -> None:
+        try:
+            from scripts.inferencers.insightface.detector import InsightFaceDetector
+
+            InsightFaceDetector()
+        except Exception:
+            super().install()
+        return None