menghanxia commited on
Commit
a534795
1 Parent(s): 302d824

modified app.py with checkpt downloading

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  import os, requests
3
  from inference import setup_model, colorize_grayscale, predict_anchors
4
 
5
- os.system("wget https://huggingface.co/menghanxia/disco/tree/main/disco-beta.pth.tar")
6
- os.rename("disco-beta.pth.tar", "./checkpoints/disco-beta.pth.tar")
7
 
8
  ## step 1: set up model
9
  device = "cpu"
 
2
  import os, requests
3
  from inference import setup_model, colorize_grayscale, predict_anchors
4
 
5
+ os.system("wget https://huggingface.co/menghanxia/disco/tree/main/disco-beta.pth.rar")
6
+ os.rename("disco-beta.pth.rar", "./checkpoints/disco-beta.pth.rar")
7
 
8
  ## step 1: set up model
9
  device = "cpu"