add spaces.gpu

app.py

@@ -15,6 +15,7 @@ def update_model_choices(task):
     model_choices = list(SAPIENS_LITE_MODELS_PATH[task.lower()].keys())
     return gr.Dropdown(choices=model_choices, value=model_choices[0] if model_choices else None)
 
+@spaces.gpu
 def process_image(input_image, task, version):
     if isinstance(input_image, np.ndarray):
         input_image = Image.fromarray(input_image)
@@ -33,6 +34,7 @@ def process_image(input_image, task, version):
     
     return result
 
+@spaces.gpu
 def process_video(input_video, task, version):
     cap = cv2.VideoCapture(input_video)
     fps = cap.get(cv2.CAP_PROP_FPS)