xizaoqu
commited on
Commit
·
beb5f3e
1
Parent(s):
10be9cf
update
Browse files
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 |
-
|
271 |
-
|
272 |
-
|
273 |
-
|
274 |
-
|
275 |
-
|
276 |
-
|
277 |
-
|
278 |
-
|
279 |
-
|
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):
|