update

app.py

@@ -200,13 +200,11 @@ def set_denoising_steps(denoising_steps, sampling_timesteps_state):
     print("set denoising steps to", worldmem.sampling_timesteps)
     return sampling_timesteps_state
 
-def update_image_and_log(keys):
+def generate(keys):
+    print("algo frame:", len(worldmem.frames))
     actions = parse_input_to_tensor(keys)
     global input_history
     global memory_curr_frame
-
-    print("algo frame:", len(worldmem.frames))
-    
     for i in range(len(actions)):
         memory_curr_frame += 1
 
@@ -370,7 +368,7 @@ with gr.Blocks(css=css) as demo:
         """
     )
     # input_box.submit(update_image_and_log, inputs=[input_box], outputs=[image_display, video_display, log_output])
-    submit_button.click(update_image_and_log, inputs=[input_box], outputs=[image_display, video_display, log_output])
+    submit_button.click(generate, inputs=[input_box], outputs=[image_display, video_display, log_output])
     reset_btn.click(reset, outputs=[log_output, image_display])
     image_display_1.select(lambda: on_image_click(SUNFLOWERS_IMAGE), outputs=image_display)
     image_display_2.select(lambda: on_image_click(DESERT_IMAGE), outputs=image_display)