jiwan-chung commited on
Commit
61a945a
1 Parent(s): 0bf81ba
Files changed (2) hide show
  1. app.py +5 -4
  2. load.py +1 -1
app.py CHANGED
@@ -7,9 +7,10 @@ import gdown
7
  from run import launch
8
 
9
  # download
10
- if not Path('./data').is_dir():
11
- url = 'https://drive.google.com/drive/folders/1hfHWDn5iXsdjB63E5zdZBAoRLWHQC3LD'
12
- gdown.download_folder(url, quiet=True, use_cookies=False, output="./data/")
 
13
 
14
  # example image from COCO data
15
  image_urls = {
@@ -21,7 +22,7 @@ for k, url in image_urls.items():
21
  output = Path(f"data/images/{k}{ext}")
22
  if not output.is_file():
23
  output.parent.mkdir(exist_ok=True)
24
- gdown.download(url, quiet=True, use_cookies=False, output=str(output))
25
  images[k] = str(output)
26
 
27
  '''
7
  from run import launch
8
 
9
  # download
10
+ if not Path('./data/esper_demo').is_dir():
11
+ url = 'https://drive.google.com/drive/folders/1pfWN4WwruOuS5uiuPyPFVB9kidlCV7kc'
12
+ gdown.download_folder(url, quiet=False, use_cookies=False, output="./data/")
13
+ assert Path('./data/esper_demo').is_dir()
14
 
15
  # example image from COCO data
16
  image_urls = {
22
  output = Path(f"data/images/{k}{ext}")
23
  if not output.is_file():
24
  output.parent.mkdir(exist_ok=True)
25
+ gdown.download(url, quiet=False, use_cookies=False, output=str(output))
26
  images[k] = str(output)
27
 
28
  '''
load.py CHANGED
@@ -15,7 +15,7 @@ log = logging.getLogger(__name__)
15
 
16
  def load_model_args(args):
17
  checkpoint = Path(args.checkpoint + '.ckpt')
18
- assert checkpoint.is_file(), f"no checkpoint file: {args.checkpoint}"
19
  args_path = Path(args.checkpoint + '.json')
20
  if args_path.is_file():
21
  with open(args_path) as f:
15
 
16
  def load_model_args(args):
17
  checkpoint = Path(args.checkpoint + '.ckpt')
18
+ assert checkpoint.is_file(), f"no checkpoint file: {checkpoint}"
19
  args_path = Path(args.checkpoint + '.json')
20
  if args_path.is_file():
21
  with open(args_path) as f: