reagvis commited on
Commit
ebb4b09
·
verified ·
1 Parent(s): bbce6aa

Update custom_tools/facesimilarity_tool.py

Browse files
custom_tools/facesimilarity_tool.py CHANGED
@@ -7,7 +7,7 @@ from insightface.app import FaceAnalysis
7
  class FaceSimilarityTool(BaseTool):
8
  default_desc = 'Uses InsightFace to evaluate face similarity between two images.'
9
 
10
- def __init__(self, threshold=0.5, use_gpu=False):
11
  super().__init__()
12
  # Initialize InsightFace
13
  self.threshold = threshold
 
7
  class FaceSimilarityTool(BaseTool):
8
  default_desc = 'Uses InsightFace to evaluate face similarity between two images.'
9
 
10
+ def __init__(self, threshold=0.01, use_gpu=False):
11
  super().__init__()
12
  # Initialize InsightFace
13
  self.threshold = threshold