doevent commited on
Commit
ca683cb
·
verified ·
1 Parent(s): 3e0636f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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 -P")
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 -P")
13
- os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/02.jpg -q -P")
14
- os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/03.jpg -q -P")
15
- os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/04.jpg -q -P")
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"