Spaces:
Sleeping
Sleeping
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.
|
| 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
|