Update app.py

app.py

@@ -43,7 +43,7 @@ def load_image(path):
     size = max(h_orig, w_orig)
     scale = 640 / size
     w, h = int(w_orig * scale), int(h_orig * scale)
-    image = cv2.resize(image, (w, h))
+    image = cv2.imread(path)
     return image
 
 
@@ -67,8 +67,8 @@ def draw_label(image,
 
 
 @torch.inference_mode()
-def predict(image, model, face_detector, device, margin=0.4, input_size=224):
-    image = cv2.imread(image, cv2.IMREAD_COLOR)[:, :, ::-1].copy()
+def predict(image_path, model, face_detector, device, margin=0.4, input_size=224):
+    image = load_image(image_path)
     image_h, image_w = image.shape[:2]
 
     # detect faces using dlib detector