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):
|