ImanAmran commited on
Commit
7107d70
·
1 Parent(s): 407b827

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
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 using Cosine Similarity and print all distances
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
- distances[user_id] = distance
73
  if distance < closest_distance:
74
  closest_distance = distance
75
  closest_user_id = user_id
76
 
77
- # Print all distances to the logs
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}\nAll Distances: {distances}"
84
  else:
85
- return f"User not recognized. Closest Distance: {closest_distance}\nAll Distances: {distances}"
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