Update inference_manager.py
Browse files- inference_manager.py +1 -1
inference_manager.py
CHANGED
|
@@ -491,7 +491,7 @@ class ModelManager:
|
|
| 491 |
raise Exception(f"face images not provided")
|
| 492 |
start = time.time()
|
| 493 |
model.base_model_pipeline.to("cuda")
|
| 494 |
-
|
| 495 |
faceid_all_embeds = []
|
| 496 |
for image in images:
|
| 497 |
face = image#cv2.imread(image)
|
|
|
|
| 491 |
raise Exception(f"face images not provided")
|
| 492 |
start = time.time()
|
| 493 |
model.base_model_pipeline.to("cuda")
|
| 494 |
+
print("extracting face...")
|
| 495 |
faceid_all_embeds = []
|
| 496 |
for image in images:
|
| 497 |
face = image#cv2.imread(image)
|