fix ipadapter face

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-11-01 08:39:03 -04:00
parent 38ccc3a7a5
commit 28efae6b2d
3 changed files with 13 additions and 9 deletions
+2 -3
View File
@@ -113,10 +113,9 @@ class APIProcess():
images = []
scores = []
with self.queue_lock:
yolo.load()
faces = yolo.predict(image)
faces = yolo.predict('face-yolo8n', image)
for face in faces:
images.append(encode_pil_to_base64(face.face))
images.append(encode_pil_to_base64(face.item))
scores.append(face.score)
shared.state.end(api=False)
return ResFace(images=images, scores=scores)