piperod commited on
Commit
f7db529
·
1 Parent(s): d2aa7b7
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -1,15 +1,17 @@
1
  import gradio as gr
2
  import os
3
-
4
 
5
  if os.getenv('SYSTEM') == 'spaces':
6
- import mim
7
 
8
- mim.uninstall('mmcv-full', confirm_yes=True)
9
- mim.install('mmcv-full==1.5.2', is_yes=True)
 
 
 
 
 
10
 
11
- subprocess.call('pip uninstall -y opencv-python'.split())
12
- subprocess.call('pip uninstall -y opencv-python-headless'.split())
13
  subprocess.call('pip install opencv-python-headless==4.5.5.64'.split())
14
 
15
  import cv2
 
1
  import gradio as gr
2
  import os
3
+ import subprocess
4
 
5
  if os.getenv('SYSTEM') == 'spaces':
 
6
 
7
+
8
+ subprocess.call('pip install -U openmim')
9
+ subprocess.call('torch==1.11.0 torchvision==0.12.0')
10
+
11
+ import mim
12
+ mim.install('mmcv>=2.0.0', confirm_yes=True)
13
+ mim.install('mmengine', is_yes=True)
14
 
 
 
15
  subprocess.call('pip install opencv-python-headless==4.5.5.64'.split())
16
 
17
  import cv2