Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -6,13 +6,13 @@ from inference import setup_model, colorize_grayscale, predict_anchors
|
|
6 |
## local | remote
|
7 |
RUN_MODE = "remote"
|
8 |
if RUN_MODE != "local":
|
9 |
-
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/disco-beta.pth.rar -q
|
10 |
os.rename("disco-beta.pth.rar", "./checkpoints/disco-beta.pth.rar")
|
11 |
## examples
|
12 |
-
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/01.jpg -q
|
13 |
-
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/02.jpg -q
|
14 |
-
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/03.jpg -q
|
15 |
-
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/04.jpg -q
|
16 |
|
17 |
## step 1: set up model
|
18 |
device = "cpu"
|
|
|
6 |
## local | remote
|
7 |
RUN_MODE = "remote"
|
8 |
if RUN_MODE != "local":
|
9 |
+
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/disco-beta.pth.rar -q")
|
10 |
os.rename("disco-beta.pth.rar", "./checkpoints/disco-beta.pth.rar")
|
11 |
## examples
|
12 |
+
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/01.jpg -q")
|
13 |
+
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/02.jpg -q")
|
14 |
+
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/03.jpg -q")
|
15 |
+
os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/04.jpg -q")
|
16 |
|
17 |
## step 1: set up model
|
18 |
device = "cpu"
|