update

app.py

@@ -184,6 +184,7 @@ memory_frames.append(load_image_as_tensor(DEFAULT_IMAGE))
 
 @spaces.GPU()
 def run_interactive(first_frame, action, first_pose, curr_frame, device):
+    global worldmem
     new_frame = worldmem.interactive(first_frame,
                                     action,
                                     first_pose, 
@@ -253,20 +254,11 @@ def reset():
     return input_history, DEFAULT_IMAGE
 
 def on_image_click(SELECTED_IMAGE):
-        global DEFAULT_IMAGE
-        DEFAULT_IMAGE = SELECTED_IMAGE
-        reset()
-        return SELECTED_IMAGE
-
-# new_frame = runner.run(
-#                     memory_frames[0],
-#                     actions[0],
-#                     poses[0],
-#                     memory_curr_frame,
-#                     device
-#                 )
-
-# print("first algo frame:", len(algo.frames))
+    global DEFAULT_IMAGE
+    DEFAULT_IMAGE = SELECTED_IMAGE
+    reset()
+    return SELECTED_IMAGE
+
 
 css = """
 h1 {