Merge pull request huggingface#39 from Ambrosiussen/master

Added missing "mediapipe" dependency for mediapipe_face_common.py
dg845 · May 22, 2023 · 9408286 · 9408286
2 parents 3c796f3 + fa7712c
commit 9408286
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/controlnet_aux/mediapipe_face/mediapipe_face_common.py b/src/controlnet_aux/mediapipe_face/mediapipe_face_common.py
@@ -1,6 +1,10 @@
 from typing import Mapping
 
-import mediapipe as mp
+try:
+    import mediapipe as mp
+except ImportError:
+    raise ImportError("The module 'mediapipe' is not installed. Please install it using the command: pip install 'mediapipe'")
+
 import numpy