xizaoqu
commited on
Commit
·
b799cfa
1
Parent(s):
4652b57
update
Browse files
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):
|