fix

app.py

@@ -78,6 +78,17 @@ if __name__ == '__main__':
         action_stride=1,
         domain='language_table',
     )
+    image = Image.open("sim/assets/langtable_prompt/frame_06.png")
+    prompt_image = np.tile(
+        np.array(image), (genie.prompt_horizon, 1, 1, 1)
+    ).astype(np.uint8)
+    prompt_action = np.zeros(
+        (genie.prompt_horizon, genie.action_stride, 2)
+    ).astype(np.float32)
+
+    genie.set_initial_state((prompt_image, prompt_action))
+    image = genie.reset()
+
     with gr.Blocks() as demo:
         with gr.Row():
             image_selector = gr.Dropdown(
@@ -95,17 +106,7 @@ if __name__ == '__main__':
             down = gr.Button("↓ Down")
             right = gr.Button("→ Right")
 
-        if not genie.allow_external_prompt and genie.gt_phys_sim is not None:
-            image = Image.open("sim/assets/langtable_prompt/frame_06.png")
-            prompt_image = np.tile(
-                np.array(image), (genie.prompt_horizon, 1, 1, 1)
-            ).astype(np.uint8)
-            prompt_action = np.zeros(
-                (genie.prompt_horizon, genie.action_stride, 2)
-            ).astype(np.float32)
-
-            genie.set_initial_state((prompt_image, prompt_action))
-            image = genie.reset()
+
 
         # Define interactions
         select_button.click(