Spaces:
Runtime error
Runtime error
Update face_analysis.py
Browse files- face_analysis.py +1 -1
face_analysis.py
CHANGED
@@ -24,7 +24,7 @@ def cluster_faces(embeddings):
|
|
24 |
return np.zeros(len(embeddings), dtype=int)
|
25 |
|
26 |
X = np.stack(embeddings)
|
27 |
-
dbscan = DBSCAN(eps=0.
|
28 |
clusters = dbscan.fit_predict(X)
|
29 |
|
30 |
if np.all(clusters == -1):
|
|
|
24 |
return np.zeros(len(embeddings), dtype=int)
|
25 |
|
26 |
X = np.stack(embeddings)
|
27 |
+
dbscan = DBSCAN(eps=0.3, min_samples=5, metric='cosine')
|
28 |
clusters = dbscan.fit_predict(X)
|
29 |
|
30 |
if np.all(clusters == -1):
|