Update app.py

app.py

@@ -6,6 +6,7 @@ import cv2
 import torch
 import facer
 from typing import Tuple
+import torchvision.transforms as transforms
 
 def process_image(input_image: np.ndarray) -> np.ndarray:
     """
@@ -21,9 +22,11 @@ def process_image(input_image: np.ndarray) -> np.ndarray:
     
     # Convert numpy array to PIL Image and back to ensure correct format
     input_pil = Image.fromarray(input_image)
+
+    transform = transforms.Compose([transforms.PILToTensor()])
     
     # Convert image to format expected by facer
-    image = facer.hwc2bchw(np.array(input_pil)).to(device=device)
+        image = facer.hwc2bchw(transform(input_pil)).to(device=device)
     
     # Initialize face detector
     face_detector = facer.face_detector('retinaface/mobilenet', device=device)