ironjr commited on
Commit
efeb5fb
1 Parent(s): 81ff5e6

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -11,6 +11,7 @@
11
  #
12
  # For permission requests, please contact robot0321@snu.ac.kr, esw0116@snu.ac.kr, namhj28@gmail.com, jarin.lee@gmail.com.
13
  import os
 
14
  import time
15
  import pathlib
16
  import shlex
@@ -24,7 +25,8 @@ example_root = os.path.join(root, 'examples')
24
  ckpt_root = os.path.join(root, 'stablediffusion')
25
 
26
  d = example_root
27
- snapshot_download(repo_id="ironjr/LucidDreamerDemo", repo_type="model", local_dir=d)
 
28
 
29
  d = os.path.join(ckpt_root, 'Blazing Drive V11m')
30
  if not os.path.exists(d):
 
11
  #
12
  # For permission requests, please contact robot0321@snu.ac.kr, esw0116@snu.ac.kr, namhj28@gmail.com, jarin.lee@gmail.com.
13
  import os
14
+ import glob
15
  import time
16
  import pathlib
17
  import shlex
 
25
  ckpt_root = os.path.join(root, 'stablediffusion')
26
 
27
  d = example_root
28
+ if len(glob.glob(os.path.join(d, '*.ply'))) < 8:
29
+ snapshot_download(repo_id="ironjr/LucidDreamerDemo", repo_type="model", local_dir=d)
30
 
31
  d = os.path.join(ckpt_root, 'Blazing Drive V11m')
32
  if not os.path.exists(d):