fix

app.py

@@ -23,16 +23,7 @@ IMAGE_DIR = "sim/assets/langtable_prompt/"
 available_images = sorted([img for img in os.listdir(IMAGE_DIR) if img.endswith(".png")])
 
 
-genie = GenieSimulator(
-    image_encoder_type='temporalvae',
-    image_encoder_ckpt='stabilityai/stable-video-diffusion-img2vid',
-    quantize=False,
-    backbone_type='stmar',
-    backbone_ckpt='data/mar_ckpt/langtable',
-    prompt_horizon=PROMPT_HORIZON,
-    action_stride=1,
-    domain='language_table',
-)
+
 
 # Helper function to reset GenieSimulator with the selected image
 def initialize_simulator(image_name):
@@ -75,6 +66,17 @@ def handle_image_selection(image_name):
 
 if __name__ == '__main__':
     with gr.Blocks() as demo:
+        genie = GenieSimulator(
+            image_encoder_type='temporalvae',
+            image_encoder_ckpt='stabilityai/stable-video-diffusion-img2vid',
+            quantize=False,
+            backbone_type='stmar',
+            backbone_ckpt='data/mar_ckpt/langtable',
+            prompt_horizon=PROMPT_HORIZON,
+            action_stride=1,
+            domain='language_table',
+        )
+
         with gr.Row():
             image_selector = gr.Dropdown(
                 choices=available_images, value=available_images[0], label="Select an Image"