Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import os
 import torch
@@ -5,11 +6,12 @@ from PIL import Image
 from inference.flovd_demo import generate_video
 import requests
 import shutil
-import os
 from huggingface_hub import snapshot_download
 
+# Get the HF token from environment variable
 hf_token = os.getenv("HF_TOKEN")
 
+# Download weights into ./ckpt
 snapshot_download(
     repo_id="roll-ai/FloVD-weights",
     repo_type="dataset",
@@ -18,6 +20,15 @@ snapshot_download(
     token=hf_token
 )
 
+# ✅ Print the directory tree of ./ckpt
+print("📂 Contents of ./ckpt directory:")
+for root, dirs, files in os.walk("./ckpt"):
+    level = root.replace("./ckpt", "").count(os.sep)
+    indent = " " * 4 * level
+    print(f"{indent}{os.path.basename(root)}/")
+    subindent = " " * 4 * (level + 1)
+    for f in files:
+        print(f"{subindent}{f}")
 # # --------- SETUP CODE: Download models if not found ---------
 # FVSM_PATH = "./ckpt/FVSM/FloVD_FVSM_Controlnet.pt"
 # OMSM_PATH = "./ckpt/OMSM"