Update app.py

app.py

@@ -9,6 +9,7 @@ except:
     
 import mmpose
 import gradio as gr
+import spaces
 from mmpose.apis import (inference_top_down_pose_model, init_pose_model,
                          vis_pose_result, process_mmdet_results)
 from mmdet.apis import inference_detector, init_detector
@@ -26,6 +27,7 @@ pose_model = init_pose_model(pose_config, pose_checkpoint, device='cuda')
 # initialize detector
 det_model = init_detector(det_config, det_checkpoint, device='cuda')
 
+@spaces.GPU
 def predict(img):
     mmdet_results = inference_detector(det_model, img)
     person_results = process_mmdet_results(mmdet_results, cat_id=1)