Update app.py

app.py

@@ -18,6 +18,12 @@ pose_checkpoint = 'hrnet_w48_coco_256x192-b9e0b3ab_20200708.pth'
 det_config = 'configs/faster_rcnn_r50_fpn_1x_coco.py'
 det_checkpoint = 'faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'
 
+@spaces.GPU
+def cache_features():
+    init_pose_model(pose_config, pose_checkpoint, device='cuda')
+    init_detector(det_config, det_checkpoint, device='cuda')
+cache_features()
+
 # initialize pose model
 pose_model = init_pose_model(pose_config, pose_checkpoint, device='cuda')
 # initialize detector