Spaces:
Build error
Build error
File size: 869 Bytes
f10370d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from .embed import get_embedding
from .utils import load_embeddings, save_embeddings
def register_new_user(face_image, name):
"""
Registers a new user by generating an embedding and persisting it.
Args:
face_image (np.array): The input BGR image frame (must contain a face).
name (str): The name of the user to register.
Returns:
bool: True on successful registration, False otherwise.
"""
embedding = get_embedding(face_image)
if embedding is not None:
data = load_embeddings()
# Store the created embedding and name
data['embeddings'].append(embedding)
data['names'].append(name)
# Persist data to {data/embeddings.pkl} [cite: 38]
return save_embeddings(data)
return False |