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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -20,7 +20,8 @@ WEIGHT_FILES = {
20
  "ckpt/OMSM/pytorch_lora_weights.safetensors": "OMSM/pytorch_lora_weights.safetensors",
21
  "ckpt/others/depth_anything_v2_metric_hypersim_vitb.pth": "others/depth_anything_v2_metric_hypersim_vitb.pth"
22
  }
23
-
 
24
  def download_weights():
25
  print("πŸ”„ Downloading model weights via huggingface_hub...")
26
  for hf_path, local_rel_path in WEIGHT_FILES.items():
@@ -36,7 +37,7 @@ def download_weights():
36
  else:
37
  print(f"βœ… Already exists: {local_path}")
38
 
39
- download_weights()
40
 
41
  # =========================================
42
  # 2. Import the FloVD generation pipeline
 
20
  "ckpt/OMSM/pytorch_lora_weights.safetensors": "OMSM/pytorch_lora_weights.safetensors",
21
  "ckpt/others/depth_anything_v2_metric_hypersim_vitb.pth": "others/depth_anything_v2_metric_hypersim_vitb.pth"
22
  }
23
+ print("")
24
+ print("Downloading model...", flush=True)
25
  def download_weights():
26
  print("πŸ”„ Downloading model weights via huggingface_hub...")
27
  for hf_path, local_rel_path in WEIGHT_FILES.items():
 
37
  else:
38
  print(f"βœ… Already exists: {local_path}")
39
 
40
+ # download_weights()
41
 
42
  # =========================================
43
  # 2. Import the FloVD generation pipeline