Spaces:
Runtime error
Runtime error
Fix
Browse files- app.py +4 -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 |
-
|
| 13 |
-
|
|
|
|
|
|
|
| 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
|