fix

app.py

@@ -23,7 +23,7 @@ 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)
@@ -34,6 +34,7 @@ def initialize_simulator(image_name, state):
     reset_image = cv2.resize(reset_image, (RES, RES))
     return Image.fromarray(reset_image)
 
+@spaces.GPU
 def model(direction, state):
     if direction == 'right':
         action = np.array([0, 0.05])
@@ -49,11 +50,13 @@ def model(direction, state):
     next_image = cv2.resize(next_image, (RES, RES))
     return Image.fromarray(next_image)
 
+@spaces.GPU
 def handle_input(direction, state):
     print(f"User clicked: {direction}")
     new_image = model(direction, state)
     return new_image
 
+@spaces.GPU
 def handle_image_selection(image_name, state):
     print(f"User selected image: {image_name}")
     return initialize_simulator(image_name, state)