Update app.py
Browse files
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 |
# =========================================
|