Update

app.py

@@ -8,6 +8,7 @@ import sys
 
 import gradio as gr
 import numpy as np
+import spaces
 import torch
 from huggingface_hub import hf_hub_download
 
@@ -61,12 +62,15 @@ def load_model(model_name: str, device: torch.device) -> FaceParser:
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-detector = RetinaFacePredictor(threshold=0.8, device=device, model=RetinaFacePredictor.get_model("mobilenet0.25"))
+detector = RetinaFacePredictor(threshold=0.8, device="cpu", model=RetinaFacePredictor.get_model("mobilenet0.25"))
+detector.device = device
+detector.net.to(device)
 
 model_names = list(WEIGHT.keys())
 models = {name: load_model(name, device=device) for name in model_names}
 
 
+@spaces.GPU
 def predict(image: np.ndarray, model_name: str, max_num_faces: int) -> np.ndarray:
     model = models[model_name]
     colormap = label_colormap(model.num_classes)
@@ -105,7 +109,6 @@ with gr.Blocks(css="style.css") as demo:
         inputs=[image, model_name, max_num_faces],
         outputs=result,
         fn=predict,
-        cache_examples=os.getenv("CACHE_EXAMPLES") == "1",
     )
 
     run_button.click(