fix

app.py

@@ -26,6 +26,7 @@ available_images = sorted([img for img in os.listdir(IMAGE_DIR) if img.endswith(
 
 
 # Helper function to reset GenieSimulator with the selected image
+@spaces.GPU
 def initialize_simulator(image_name):
     image_path = os.path.join(IMAGE_DIR, image_name)
     image = Image.open(image_path)
@@ -37,6 +38,7 @@ def initialize_simulator(image_name):
     return Image.fromarray(reset_image)
 
 # Example model: takes a direction and returns a random image
+@spaces.GPU
 def model(direction: str):
     if direction == 'right':
         action = np.array([0, 0.05])
@@ -60,13 +62,13 @@ def handle_input(direction):
     return new_image
 
 # Gradio function to handle image selection
+@spaces.GPU
 def handle_image_selection(image_name):
     print(f"User selected image: {image_name}")
     return initialize_simulator(image_name)
 
 if __name__ == '__main__':
-    with gr.Blocks() as demo:
-        genie = GenieSimulator(
+    genie = GenieSimulator(
             image_encoder_type='temporalvae',
             image_encoder_ckpt='stabilityai/stable-video-diffusion-img2vid',
             quantize=False,
@@ -77,6 +79,7 @@ if __name__ == '__main__':
             domain='language_table',
         )
 
+    with gr.Blocks() as demo:
         with gr.Row():
             image_selector = gr.Dropdown(
                 choices=available_images, value=available_images[0], label="Select an Image"