test1444 commited on
Commit
d024259
1 Parent(s): 0582a98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -1,6 +1,14 @@
1
  import torch, torchvision
 
 
 
 
 
 
 
 
 
2
  import mmpose
3
- from mmcv.ops import get_compiling_cuda_version, get_compiler_version
4
  import gradio as gr
5
  import cv2
6
  from mmpose.apis import (inference_top_down_pose_model, init_pose_model,
 
1
  import torch, torchvision
2
+ import sys
3
+ import subprocess
4
+ try:
5
+ from mmcv.ops import get_compiling_cuda_version, get_compiler_version
6
+ except:
7
+ subprocess.check_call([sys.executable, '-m', 'mim', 'install',
8
+ 'mmcv-full==1.5.0'])
9
+ from mmcv.ops import get_compiling_cuda_version, get_compiler_version
10
+
11
  import mmpose
 
12
  import gradio as gr
13
  import cv2
14
  from mmpose.apis import (inference_top_down_pose_model, init_pose_model,