nsfwalex commited on
Commit
316f7d7
·
verified ·
1 Parent(s): a0b5cec

Update inference_manager.py

Browse files
Files changed (1) hide show
  1. inference_manager.py +1 -1
inference_manager.py CHANGED
@@ -445,7 +445,7 @@ class ModelManager:
445
 
446
  faceid_all_embeds = []
447
  for image in images:
448
- face = cv2.imread(image)
449
  faces = self.app.get(face)
450
  faceid_embed = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
451
  faceid_all_embeds.append(faceid_embed)
 
445
 
446
  faceid_all_embeds = []
447
  for image in images:
448
+ face = image#cv2.imread(image)
449
  faces = self.app.get(face)
450
  faceid_embed = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
451
  faceid_all_embeds.append(faceid_embed)