xizaoqu commited on
Commit
beb5f3e
·
1 Parent(s): 10be9cf
Files changed (1) hide show
  1. app.py +11 -16
app.py CHANGED
@@ -199,7 +199,7 @@ def run_interactive(first_frame, action, first_pose, curr_frame, device, self_fr
199
  self_poses=self_poses,
200
  self_memory_c2w=self_memory_c2w,
201
  self_frame_idx=self_frame_idx)
202
- return new_frame, self_frames, self_poses, self_memory_c2w, self_frame_idx
203
 
204
  def set_denoising_steps(denoising_steps, sampling_timesteps_state):
205
  worldmem.sampling_timesteps = denoising_steps
@@ -267,21 +267,16 @@ def reset():
267
  memory_curr_frame = 0
268
  input_history = ""
269
 
270
- print("before reset")
271
- try:
272
- _, self_frames, self_poses, self_memory_c2w, self_frame_idx = run_interactive(memory_frames[0],
273
- actions[0],
274
- poses[0],
275
- memory_curr_frame,
276
- device=device,
277
- self_frames=self_frames,
278
- self_poses=self_poses,
279
- self_memory_c2w=self_memory_c2w,
280
- self_frame_idx=self_frame_idx)
281
- except Exception as e:
282
- print("Error during reset:", e)
283
- pass
284
- print("in reset")
285
  return input_history, DEFAULT_IMAGE
286
 
287
  def on_image_click(SELECTED_IMAGE):
 
199
  self_poses=self_poses,
200
  self_memory_c2w=self_memory_c2w,
201
  self_frame_idx=self_frame_idx)
202
+ return new_frame # , self_frames, self_poses, self_memory_c2w, self_frame_idx
203
 
204
  def set_denoising_steps(denoising_steps, sampling_timesteps_state):
205
  worldmem.sampling_timesteps = denoising_steps
 
267
  memory_curr_frame = 0
268
  input_history = ""
269
 
270
+ _, = run_interactive(memory_frames[0],
271
+ actions[0],
272
+ poses[0],
273
+ memory_curr_frame,
274
+ device=device,
275
+ self_frames=self_frames,
276
+ self_poses=self_poses,
277
+ self_memory_c2w=self_memory_c2w,
278
+ self_frame_idx=self_frame_idx)
279
+
 
 
 
 
 
280
  return input_history, DEFAULT_IMAGE
281
 
282
  def on_image_click(SELECTED_IMAGE):