Add error msg for failed import

shubhamgarg21 · Oct 19, 2023 · cca3422 · cca3422
1 parent c53366b
commit cca3422
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/deepface/detectors/FastMtcnnWrapper.py b/deepface/detectors/FastMtcnnWrapper.py
@@ -6,7 +6,11 @@
 
 def build_model():
     # Optional dependency
-    from facenet_pytorch import MTCNN as fast_mtcnn
+    try:
+        from facenet_pytorch import MTCNN as fast_mtcnn
+    except ModuleNotFoundError:
+        print("This is an optional detector, ensure the library is installed. \
+              Please install using 'pip install facenet-pytorch' ")
 
 
     face_detector = fast_mtcnn(image_size=160,