Update app.py

app.py

@@ -1,12 +1,17 @@
+import spaces
 import torch, torchvision
 import sys
 # sys.path.insert(0, 'test_mmpose/')
-try:
-    from mmcv.ops import get_compiling_cuda_version, get_compiler_version
-except:
-    import mim
-    mim.install('mmcv-full==1.5.0')
-    
+
+@spaces.GPU
+def prepare():
+    try:
+        from mmcv.ops import get_compiling_cuda_version, get_compiler_version
+    except:
+        import mim
+        mim.install('mmcv-full==1.5.0')
+prepare()    
+
 import mmpose
 import gradio as gr
 
@@ -27,6 +32,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)
@@ -121,7 +127,7 @@ article = ""
 demo = gr.Interface(fn=predict,
                     inputs=gr.Image(), 
                     outputs=[gr.Image(label='Prediction'), gr.Image(label='Poses')], 
-                    examples=example_list, 
+                    #examples=example_list, 
                     title=title,
                     description=description,
                     article=article)