cpu?

app copy.py

@@ -23,7 +23,6 @@ IMAGE_DIR = "sim/assets/langtable_prompt/"
 available_images = sorted([img for img in os.listdir(IMAGE_DIR) if img.endswith(".png")])
 
 
-@spaces.GPU
 def initialize_simulator(image_name, state):
     image_path = os.path.join(IMAGE_DIR, image_name)
     image = Image.open(image_path)

app.py

@@ -89,8 +89,6 @@ if __name__ == '__main__':
         ).astype(np.float32)
         genie.set_initial_state((prompt_image, prompt_action))
         image = genie.reset()
-
-
         with gr.Row():
             image_selector = gr.Dropdown(
                 choices=available_images, value=available_images[0], label="Select an Image"

sim/simulator.py

@@ -468,7 +468,6 @@ class GenieSimulator(LearnedSimulator):
         image = np.clip(image, 0, 255).astype(np.uint8)
         return image
 
-    @spaces.GPU
     def reset(self) -> np.ndarray:
         # if ground truth physics simulator is provided,
         # return the the side-by-side concatenated image