adymaharana commited on
Commit
da4893d
1 Parent(s): e217e55
Files changed (2) hide show
  1. app.py +4 -1
  2. ckpt/25.pth +3 -0
app.py CHANGED
@@ -73,7 +73,10 @@ def main(args):
73
  #model_url = 'https://drive.google.com/u/1/uc?id=1lJ6zMZ6qTvFu6H35-VEdFlN13MMslivJ&export=download'
74
  png_url = 'https://drive.google.com/u/1/uc?id=1C33A1IzSHDPoQ4QBsgFWbF61QWaAxRo_&export=download'
75
 
76
- gdown.download(model_url, quiet=False, use_cookies=False, output="./ckpt/25.pth")
 
 
 
77
  gdown.download(png_url, quiet=True, use_cookies=False, output="demo_pororo_good.png")
78
 
79
  if args.debug:
73
  #model_url = 'https://drive.google.com/u/1/uc?id=1lJ6zMZ6qTvFu6H35-VEdFlN13MMslivJ&export=download'
74
  png_url = 'https://drive.google.com/u/1/uc?id=1C33A1IzSHDPoQ4QBsgFWbF61QWaAxRo_&export=download'
75
 
76
+ #if not os.path.exists("./ckpt/25.pth"):
77
+ # gdown.download(model_url, quiet=False, use_cookies=False, output="./ckpt/25.pth")
78
+ # print("Downloaded checkpoint")
79
+ assert os.path.exists("./ckpt/25.pth")
80
  gdown.download(png_url, quiet=True, use_cookies=False, output="demo_pororo_good.png")
81
 
82
  if args.debug:
ckpt/25.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c6cf13b9300cbc23c1de8fdd6267b562d267a42ca3aa83a9b664b04c7881906
3
+ size 16147881312