Update

app.py

@@ -10,6 +10,7 @@ import urllib.request
 import cv2
 import gradio as gr
 import numpy as np
+import spaces
 import torch
 from huggingface_hub import hf_hub_download
 
@@ -50,12 +51,15 @@ if is_lfs_pointer_file(lfs_model_path):
 
 def load_model(model_name: str, threshold: float, device: torch.device) -> RetinaFacePredictor | S3FDPredictor:
     if model_name == "s3fd":
-        model = S3FDPredictor(threshold=threshold, device=device)
+        model = S3FDPredictor(threshold=threshold, device="cpu")
+        model.device = device
+        model.net.device = device
+        model.net.to(device)
     else:
         model_name = model_name.replace("retinaface_", "")
-        model = RetinaFacePredictor(
-            threshold=threshold, device=device, model=RetinaFacePredictor.get_model(model_name)
-        )
+        model = RetinaFacePredictor(threshold=threshold, device="cpu", model=RetinaFacePredictor.get_model(model_name))
+        model.device = device
+        model.net.to(device)
     return model
 
 
@@ -68,6 +72,7 @@ model_names = [
 detectors = {name: load_model(name, threshold=0.8, device=device) for name in model_names}
 
 
+@spaces.GPU
 def detect(image: np.ndarray, model_name: str, face_score_threshold: float) -> np.ndarray:
     model = detectors[model_name]
     model.threshold = face_score_threshold