update assets

app.py

@@ -270,25 +270,24 @@ def generate(keys, input_history, memory_frames, self_frames, self_actions, self
 
     temporal_video_path = tempfile.NamedTemporaryFile(suffix='.mp4').name
     save_video(out_video, temporal_video_path)
-
-    
-    now = datetime.now()
-    folder_name = now.strftime("%Y-%m-%d_%H-%M-%S")
-    folder_path = os.path.join("/mnt/xiaozeqi/worldmem/output_material", folder_name)
-    os.makedirs(folder_path, exist_ok=True)
     input_history += keys
 
-    data_dict = {
-        "input_history": input_history,
-        "memory_frames": memory_frames,
-        "self_frames": self_frames,
-        "self_actions": self_actions,
-        "self_poses": self_poses,
-        "self_memory_c2w": self_memory_c2w,
-        "self_frame_idx": self_frame_idx,
-    }
-
-    np.savez(os.path.join(folder_path, "data_bundle.npz"), **data_dict)
+    
+    # now = datetime.now()
+    # folder_name = now.strftime("%Y-%m-%d_%H-%M-%S")
+    # folder_path = os.path.join("/mnt/xiaozeqi/worldmem/output_material", folder_name)
+    # os.makedirs(folder_path, exist_ok=True)
+    # data_dict = {
+    #     "input_history": input_history,
+    #     "memory_frames": memory_frames,
+    #     "self_frames": self_frames,
+    #     "self_actions": self_actions,
+    #     "self_poses": self_poses,
+    #     "self_memory_c2w": self_memory_c2w,
+    #     "self_frame_idx": self_frame_idx,
+    # }
+
+    # np.savez(os.path.join(folder_path, "data_bundle.npz"), **data_dict)
 
     return out_video[-1], temporal_video_path, input_history, memory_frames, self_frames, self_actions, self_poses, self_memory_c2w, self_frame_idx
 
@@ -319,7 +318,7 @@ def on_image_click(selected_image):
 
 def set_memory(examples_case, image_display, log_output, slider_denoising_step,  slider_context_length, slider_memory_length):
     if examples_case == '1':
-        data_bundle = np.load("/mnt/xiaozeqi/worldmem/output_material/2025-04-11_16-01-49/data_bundle.npz")
+        data_bundle = np.load("assets/examples/case1.npz")
         input_history = data_bundle['input_history'].item()
         memory_frames = data_bundle['memory_frames']
         self_frames = data_bundle['self_frames']
@@ -328,7 +327,7 @@ def set_memory(examples_case, image_display, log_output, slider_denoising_step,
         self_memory_c2w = data_bundle['self_memory_c2w']
         self_frame_idx = data_bundle['self_frame_idx']
     elif examples_case == '2':
-        data_bundle = np.load("/mnt/xiaozeqi/worldmem/output_material/2025-04-12_10-42-04/data_bundle.npz")
+        data_bundle = np.load("assets/examples/case2.npz")
         input_history = data_bundle['input_history'].item()
         memory_frames = data_bundle['memory_frames']
         self_frames = data_bundle['self_frames']
@@ -337,7 +336,7 @@ def set_memory(examples_case, image_display, log_output, slider_denoising_step,
         self_memory_c2w = data_bundle['self_memory_c2w']
         self_frame_idx = data_bundle['self_frame_idx']
     elif examples_case == '3':
-        data_bundle = np.load("/mnt/xiaozeqi/worldmem/output_material/2025-04-12_10-56-57/data_bundle.npz")
+        data_bundle = np.load("assets/examples/case3.npz")
         input_history = data_bundle['input_history'].item()
         memory_frames = data_bundle['memory_frames']
         self_frames = data_bundle['self_frames']
@@ -346,7 +345,7 @@ def set_memory(examples_case, image_display, log_output, slider_denoising_step,
         self_memory_c2w = data_bundle['self_memory_c2w']
         self_frame_idx = data_bundle['self_frame_idx']
     elif examples_case == '4':
-        data_bundle = np.load("/mnt/xiaozeqi/worldmem/output_material/2025-04-11_16-07-19/data_bundle.npz")
+        data_bundle = np.load("assets/examples/case4.npz")
         input_history = data_bundle['input_history'].item()
         memory_frames = data_bundle['memory_frames']
         self_frames = data_bundle['self_frames']

assets/examples/case1.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:38600ac4addd4546591f16c3cbed3e37ea3396286b3aee118938494f6c0527dd
+size 201735738

assets/examples/case2.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a915bfd1a0ef18734f3b670b228eb4f8346736205fb571d4caa30d0a3cb919e4
+size 277385958

assets/examples/case3.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f9d09230c9d3e5fca2f4a2f0a34823c98c644c7920e9a132ab631b72ad442fa7
+size 198933878

assets/examples/case4.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:12cea63662d99d6e6349a602b2b5b5044de29b5176e7727e1e1eec528d2a56de
+size 179320858