change_loading_options

app.py

@@ -43,11 +43,22 @@ def load_large_files(relative_path):
     # Temporary download in Space
     file_path = "/tmp/" + relative_path.split('/')[-1]
     
-    response = requests.get(url)
-    with open(file_path, "wb") as f:
-        f.write(response.content)
+    response = requests.get(url, stream=True)
+    response.raise_for_status()
+    total_size = int(response.headers.get("content-length", 0))
+    with open(file_path, "wb") as file, tqdm(
+        desc="Downloading",
+        total=total_size,
+        unit="B",
+        unit_scale=True,
+        unit_divisor=1024,
+    ) as bar:
+        for chunk in response.iter_content(chunk_size=8192):
+            file.write(chunk)
+            bar.update(len(chunk))
+    print(f"File downloaded successfully and saved as: {file_path}")
     return file_path
-    
+
 def motion_temporal_filter(motion, sigma=1):
     motion = motion.reshape(motion.shape[0], -1)
     for i in range(motion.shape[1]):
@@ -104,7 +115,6 @@ else:
     load_file_list = {}
     for key in src_file_list.keys():
         load_file_list[key] = load_large_files(src_file_list[key])
-    print(load_file_list)
 
 model_lmm = create_lmm()
 model_imagebind = imagebind_huge(pretrained=True, ckpt_path=load_file_list["imagebind"])