fix lib version
Browse files
model.py
CHANGED
|
@@ -7,15 +7,15 @@ import subprocess
|
|
| 7 |
import sys
|
| 8 |
import json
|
| 9 |
|
| 10 |
-
|
| 11 |
-
|
| 12 |
|
| 13 |
-
|
| 14 |
-
|
| 15 |
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
|
| 20 |
import huggingface_hub
|
| 21 |
import numpy as np
|
|
|
|
| 7 |
import sys
|
| 8 |
import json
|
| 9 |
|
| 10 |
+
if os.getenv("SYSTEM") == "spaces":
|
| 11 |
+
import mim
|
| 12 |
|
| 13 |
+
mim.uninstall("mmcv-full", confirm_yes=True)
|
| 14 |
+
mim.install("mmcv-full==1.5.0", is_yes=True)
|
| 15 |
|
| 16 |
+
subprocess.run(shlex.split("pip uninstall -y opencv-python"))
|
| 17 |
+
subprocess.run(shlex.split("pip uninstall -y opencv-python-headless"))
|
| 18 |
+
subprocess.run(shlex.split("pip install opencv-python-headless==4.5.5.64"))
|
| 19 |
|
| 20 |
import huggingface_hub
|
| 21 |
import numpy as np
|