Spaces:
Running
Running
Update
Browse files
app.py
CHANGED
@@ -2,7 +2,28 @@
|
|
2 |
|
3 |
from __future__ import annotations
|
4 |
|
|
|
5 |
import pathlib
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
|
7 |
import gradio as gr
|
8 |
|
|
|
2 |
|
3 |
from __future__ import annotations
|
4 |
|
5 |
+
import os
|
6 |
import pathlib
|
7 |
+
import shlex
|
8 |
+
import subprocess
|
9 |
+
|
10 |
+
if os.getenv("SYSTEM") == "spaces":
|
11 |
+
subprocess.run(shlex.split("pip install click==7.1.2"))
|
12 |
+
subprocess.run(shlex.split("pip install typer==0.9.4"))
|
13 |
+
|
14 |
+
import mim
|
15 |
+
|
16 |
+
mim.uninstall("mmcv-full", confirm_yes=True)
|
17 |
+
mim.install("mmcv-full==1.5.0", is_yes=True)
|
18 |
+
|
19 |
+
subprocess.run(shlex.split("pip uninstall -y opencv-python"))
|
20 |
+
subprocess.run(shlex.split("pip uninstall -y opencv-python-headless"))
|
21 |
+
subprocess.run(shlex.split("pip install opencv-python-headless==4.8.0.74"))
|
22 |
+
|
23 |
+
with open("patch") as f:
|
24 |
+
subprocess.run(shlex.split("patch -p1"), cwd="CBNetV2", stdin=f)
|
25 |
+
subprocess.run("mv palette.py CBNetV2/mmdet/core/visualization/".split())
|
26 |
+
|
27 |
|
28 |
import gradio as gr
|
29 |
|
model.py
CHANGED
@@ -1,25 +1,8 @@
|
|
1 |
from __future__ import annotations
|
2 |
|
3 |
-
import os
|
4 |
import pathlib
|
5 |
-
import shlex
|
6 |
-
import subprocess
|
7 |
import sys
|
8 |
|
9 |
-
if os.getenv("SYSTEM") == "spaces":
|
10 |
-
import mim
|
11 |
-
|
12 |
-
mim.uninstall("mmcv-full", confirm_yes=True)
|
13 |
-
mim.install("mmcv-full==1.5.0", is_yes=True)
|
14 |
-
|
15 |
-
subprocess.run(shlex.split("pip uninstall -y opencv-python"))
|
16 |
-
subprocess.run(shlex.split("pip uninstall -y opencv-python-headless"))
|
17 |
-
subprocess.run(shlex.split("pip install opencv-python-headless==4.8.0.74"))
|
18 |
-
|
19 |
-
with open("patch") as f:
|
20 |
-
subprocess.run(shlex.split("patch -p1"), cwd="CBNetV2", stdin=f)
|
21 |
-
subprocess.run("mv palette.py CBNetV2/mmdet/core/visualization/".split())
|
22 |
-
|
23 |
import numpy as np
|
24 |
import torch
|
25 |
import torch.nn as nn
|
|
|
1 |
from __future__ import annotations
|
2 |
|
|
|
3 |
import pathlib
|
|
|
|
|
4 |
import sys
|
5 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
import numpy as np
|
7 |
import torch
|
8 |
import torch.nn as nn
|