hysts HF staff commited on
Commit
6d966e1
1 Parent(s): 4895fef
Files changed (2) hide show
  1. app.py +4 -2
  2. requirements.txt +1 -0
app.py CHANGED
@@ -9,8 +9,10 @@ import subprocess
9
  import tarfile
10
 
11
  if os.getenv('SYSTEM') == 'spaces':
12
- subprocess.call('pip uninstall -y mmcv-full'.split())
13
- subprocess.call('pip install mmcv-full==1.5.2'.split())
 
 
14
 
15
  subprocess.call('pip uninstall -y opencv-python'.split())
16
  subprocess.call('pip uninstall -y opencv-python-headless'.split())
 
9
  import tarfile
10
 
11
  if os.getenv('SYSTEM') == 'spaces':
12
+ import mim
13
+
14
+ mim.uninstall('mmcv-full', confirm_yes=True)
15
+ mim.install('mmcv-full==1.5.2', is_yes=True)
16
 
17
  subprocess.call('pip uninstall -y opencv-python'.split())
18
  subprocess.call('pip uninstall -y opencv-python-headless'.split())
requirements.txt CHANGED
@@ -2,5 +2,6 @@ mmcv-full==1.5.2
2
  mmdet==2.25.0
3
  numpy==1.22.4
4
  opencv-python-headless==4.5.5.64
 
5
  torch==1.11.0
6
  torchvision==0.12.0
 
2
  mmdet==2.25.0
3
  numpy==1.22.4
4
  opencv-python-headless==4.5.5.64
5
+ openmim==0.1.5
6
  torch==1.11.0
7
  torchvision==0.12.0