This commit is contained in:
Nicolas Mowen 2025-03-19 08:39:30 -06:00
parent 98a326f025
commit ea78cced54
2 changed files with 2 additions and 3 deletions

View File

@ -207,7 +207,7 @@ async def recognize_face(request: Request, file: UploadFile):
)
context: EmbeddingsContext = request.app.embeddings
result = context.recognize_face(name, await file.read())
result = context.recognize_face(await file.read())
return JSONResponse(
status_code=200 if result.get("success", True) else 400,
content=result,

View File

@ -197,11 +197,10 @@ class EmbeddingsContext:
},
)
def recognize_face(self, face_name: str, image_data: bytes) -> dict[str, any]:
def recognize_face(self, image_data: bytes) -> dict[str, any]:
return self.requestor.send_data(
EmbeddingsRequestEnum.recognize_face.value,
{
"face_name": face_name,
"image": base64.b64encode(image_data).decode("ASCII"),
},
)