pjoshi15 commited on
Commit
794e6c4
·
verified ·
1 Parent(s): bda0065

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def process_image(input_image: np.ndarray) -> np.ndarray:
26
  transform = transforms.Compose([transforms.PILToTensor()])
27
 
28
  # Convert image to format expected by facer
29
- image = facer.hwc2bchw(transform(input_pil)).to(device=device)
30
 
31
  # Initialize face detector
32
  face_detector = facer.face_detector('retinaface/mobilenet', device=device)
 
26
  transform = transforms.Compose([transforms.PILToTensor()])
27
 
28
  # Convert image to format expected by facer
29
+ image = facer.hwc2bchw(transform(input_pil).permute(1, 2, 0)).to(device=device)
30
 
31
  # Initialize face detector
32
  face_detector = facer.face_detector('retinaface/mobilenet', device=device)