Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -59,30 +59,26 @@ def register_user(image, user_id):
|
|
| 59 |
except Exception as e:
|
| 60 |
return f"Error during registration: {str(e)}"
|
| 61 |
|
| 62 |
-
# Recognize user
|
| 63 |
def recognize_user(image):
|
| 64 |
try:
|
| 65 |
new_embedding = generate_embedding(image)
|
| 66 |
-
distances = {}
|
| 67 |
closest_user_id = None
|
| 68 |
closest_distance = float('inf')
|
| 69 |
|
| 70 |
for user_id, embedding in zip(user_ids, user_embeddings):
|
| 71 |
distance = cosine(new_embedding, embedding)
|
| 72 |
-
|
| 73 |
if distance < closest_distance:
|
| 74 |
closest_distance = distance
|
| 75 |
closest_user_id = user_id
|
| 76 |
|
| 77 |
-
|
| 78 |
-
for user_id, distance in distances.items():
|
| 79 |
-
print(f"Distance to User {user_id}: {distance}")
|
| 80 |
|
| 81 |
-
# Return the closest user and all distances
|
| 82 |
if closest_distance <= RECOGNITION_THRESHOLD:
|
| 83 |
-
return f"Recognized User: {closest_user_id}, Distance: {closest_distance}
|
| 84 |
else:
|
| 85 |
-
return f"User not recognized. Closest Distance: {closest_distance}
|
| 86 |
except Exception as e:
|
| 87 |
return f"Error during recognition: {str(e)}"
|
| 88 |
|
|
|
|
| 59 |
except Exception as e:
|
| 60 |
return f"Error during registration: {str(e)}"
|
| 61 |
|
| 62 |
+
# Recognize user
|
| 63 |
def recognize_user(image):
|
| 64 |
try:
|
| 65 |
new_embedding = generate_embedding(image)
|
|
|
|
| 66 |
closest_user_id = None
|
| 67 |
closest_distance = float('inf')
|
| 68 |
|
| 69 |
for user_id, embedding in zip(user_ids, user_embeddings):
|
| 70 |
distance = cosine(new_embedding, embedding)
|
| 71 |
+
print(f"Distance for {user_id}: {distance}") # Debug: Print distances for each user
|
| 72 |
if distance < closest_distance:
|
| 73 |
closest_distance = distance
|
| 74 |
closest_user_id = user_id
|
| 75 |
|
| 76 |
+
print(f"Min distance: {closest_distance}") # Debug: Print minimum distance
|
|
|
|
|
|
|
| 77 |
|
|
|
|
| 78 |
if closest_distance <= RECOGNITION_THRESHOLD:
|
| 79 |
+
return f"Recognized User: {closest_user_id}, Distance: {closest_distance}"
|
| 80 |
else:
|
| 81 |
+
return f"User not recognized. Closest Distance: {closest_distance}"
|
| 82 |
except Exception as e:
|
| 83 |
return f"Error during recognition: {str(e)}"
|
| 84 |
|