roll-ai commited on
Commit
69369b0
Β·
verified Β·
1 Parent(s): e7b3b73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -39,6 +39,18 @@ def download_weights():
39
 
40
  # download_weights()
41
 
 
 
 
 
 
 
 
 
 
 
 
 
42
  # =========================================
43
  # 2. Import the FloVD generation pipeline
44
  # =========================================
 
39
 
40
  # download_weights()
41
 
42
+ def print_ckpt_structure(base_path="ckpt"):
43
+ print(f"πŸ“‚ Listing structure of: {base_path}", flush=True)
44
+ for root, dirs, files in os.walk(base_path):
45
+ level = root.replace(base_path, '').count(os.sep)
46
+ indent = ' ' * 2 * level
47
+ print(f"{indent}πŸ“ {os.path.basename(root)}/", flush=True)
48
+ sub_indent = ' ' * 2 * (level + 1)
49
+ for f in files:
50
+ print(f"{sub_indent}πŸ“„ {f}", flush=True)
51
+
52
+ # Call it
53
+ print_ckpt_structure()
54
  # =========================================
55
  # 2. Import the FloVD generation pipeline
56
  # =========================================