Update app.py

app.py

@@ -9,9 +9,7 @@ mim.install('mmcv-full==1.5.0')
 import mmpose
 import gradio as gr
 
-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
+
 from PIL import Image
 import cv2
 import numpy as np
@@ -23,10 +21,15 @@ det_checkpoint = 'faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'
 
 @spaces.GPU
 def cache_features():
+    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
     init_pose_model(pose_config, pose_checkpoint, device='cuda')
     init_detector(det_config, det_checkpoint, device='cuda')
 cache_features()
-
+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
 # initialize pose model
 pose_model = init_pose_model(pose_config, pose_checkpoint, device='cuda')
 # initialize detector