xizaoqu commited on
Commit
b799cfa
·
1 Parent(s): 4652b57
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -199,6 +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):
@@ -267,6 +268,7 @@ def reset():
267
  memory_curr_frame = 0
268
  input_history = ""
269
 
 
270
  _, self_frames, self_poses, self_memory_c2w, self_frame_idx = run_interactive(memory_frames[0],
271
  actions[0],
272
  poses[0],
@@ -277,6 +279,7 @@ def reset():
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):
 
199
  self_poses=self_poses,
200
  self_memory_c2w=self_memory_c2w,
201
  self_frame_idx=self_frame_idx)
202
+ print("in run_interactive")
203
  return new_frame, self_frames, self_poses, self_memory_c2w, self_frame_idx
204
 
205
  def set_denoising_steps(denoising_steps, sampling_timesteps_state):
 
268
  memory_curr_frame = 0
269
  input_history = ""
270
 
271
+ print("before reset")
272
  _, self_frames, self_poses, self_memory_c2w, self_frame_idx = run_interactive(memory_frames[0],
273
  actions[0],
274
  poses[0],
 
279
  self_memory_c2w=self_memory_c2w,
280
  self_frame_idx=self_frame_idx)
281
 
282
+ print("in reset")
283
  return input_history, DEFAULT_IMAGE
284
 
285
  def on_image_click(SELECTED_IMAGE):