update

app.py

@@ -188,7 +188,7 @@ self_memory_c2w = None
 self_frame_idx = None
 
 
-@spaces.GPU(duration=50)
+@spaces.GPU()
 def run_interactive(first_frame, action, first_pose, curr_frame, device, self_frames, self_poses, self_memory_c2w, self_frame_idx):
     new_frame, self_frames, self_poses, self_memory_c2w, self_frame_idx = worldmem.interactive(first_frame,
                                     action,
@@ -199,12 +199,6 @@ def run_interactive(first_frame, action, first_pose, curr_frame, device, self_fr
                                     self_poses=self_poses,
                                     self_memory_c2w=self_memory_c2w,
                                     self_frame_idx=self_frame_idx)
-    print("in run_interactive")
-    print(new_frame.device)
-    print(self_frames.device)
-    print(self_poses.device)
-    print(self_memory_c2w.device)
-    print(self_frame_idx.device)
     return new_frame, self_frames, self_poses, self_memory_c2w, self_frame_idx
 
 def set_denoising_steps(denoising_steps, sampling_timesteps_state):
@@ -274,16 +268,19 @@ def reset():
     input_history = ""
 
     print("before reset")
-    _, self_frames, self_poses, self_memory_c2w, self_frame_idx = run_interactive(memory_frames[0],
-                                actions[0],
-                                poses[0],
-                                memory_curr_frame,
-                                device=device,
-                                self_frames=self_frames,
-                                self_poses=self_poses,
-                                self_memory_c2w=self_memory_c2w,
-                                self_frame_idx=self_frame_idx)
-    
+    try:
+        _, self_frames, self_poses, self_memory_c2w, self_frame_idx = run_interactive(memory_frames[0],
+                                    actions[0],
+                                    poses[0],
+                                    memory_curr_frame,
+                                    device=device,
+                                    self_frames=self_frames,
+                                    self_poses=self_poses,
+                                    self_memory_c2w=self_memory_c2w,
+                                    self_frame_idx=self_frame_idx)
+    except Exception as e:
+        print("Error during reset:", e)
+        pass
     print("in reset")
     return input_history, DEFAULT_IMAGE