roll back load

visualizer_drag_gradio.py

@@ -225,8 +225,7 @@ with gr.Blocks() as app:
     })
 
     # init image
-    # global_state = init_images(global_state)
-    app.load(lambda state: init_images(state), global_state, global_state)
+    global_state = init_images(global_state)
     with gr.Row():
 
         with gr.Row():
@@ -245,7 +244,6 @@ with gr.Blocks() as app:
                             choices=list(valid_checkpoints_dict.keys()),
                             label="Pretrained Model",
                             value=init_pkl,
-                            interactive=not IS_SPACE
                         )
 
                 # Latent
@@ -389,13 +387,12 @@ with gr.Blocks() as app:
 
         return global_state, global_state["images"]['image_show']
 
-    if not IS_SPACE:
-        form_pretrained_dropdown.change(
-            on_change_pretrained_dropdown,
-            inputs=[form_pretrained_dropdown, global_state],
-            outputs=[global_state, form_image],
-            queue=False,
-        )
+    form_pretrained_dropdown.change(
+        on_change_pretrained_dropdown,
+        inputs=[form_pretrained_dropdown, global_state],
+        outputs=[global_state, form_image],
+        queue=True,
+    )
 
     def on_click_reset_image(global_state):
         """Reset image to the original one and clear all states