menghanxia commited on
Commit
565e612
1 Parent(s): a534795

change checkpt path to '.../resolve/..'

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ 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.rar")
6
  os.rename("disco-beta.pth.rar", "./checkpoints/disco-beta.pth.rar")
7
 
8
  ## step 1: set up model
@@ -53,7 +53,7 @@ with demo:
53
  outputs=Image_output)
54
  ## guiline
55
  gr.Markdown(value="""
56
- **Usage Guideline**
57
  1. upload your image;
58
  2. Set up the arguments: "Num. of anchors" and "Colorization resolution";
59
  3. Run the colorization (two modes supported):
 
2
  import os, requests
3
  from inference import setup_model, colorize_grayscale, predict_anchors
4
 
5
+ os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/disco-beta.pth.rar")
6
  os.rename("disco-beta.pth.rar", "./checkpoints/disco-beta.pth.rar")
7
 
8
  ## step 1: set up model
 
53
  outputs=Image_output)
54
  ## guiline
55
  gr.Markdown(value="""
56
+ **Guideline**
57
  1. upload your image;
58
  2. Set up the arguments: "Num. of anchors" and "Colorization resolution";
59
  3. Run the colorization (two modes supported):