Update db.py

resolves #11
shhossain · Aug 9, 2024 · 0652e4f · 0652e4f
1 parent 1feaf7a
commit 0652e4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/facedb/db.py b/facedb/db.py
@@ -46,7 +46,7 @@ def load_module(module: Literal["deepface", "face_recognition"]):
             if DeepFace is None:
                 try:
                     from deepface import DeepFace
-                    from deepface.commons import distance as deepface_distance
+                    from deepface.modules import verification as deepface_distance
                 except ImportError:
                     raise ImportError(
                         "Please install `deepface` to use `deepface` module."
@@ -371,7 +371,7 @@ def get_threshold(self) -> Tuple[str, str, float]:
             if metric == "euclidean" and self.l2_normalization:
                 metric = "euclidean_l2"
 
-            threshold = deepface_distance.findThreshold(  # type: ignore
+            threshold = deepface_distance.find_threshold(  # type: ignore
                 self.deepface_model_name, metric
             )
             return "le", "negative", threshold