Try decorate

app.py

@@ -3,6 +3,7 @@ import numpy as np
 from PIL import Image
 import cv2
 from sim.simulator import GenieSimulator
+import spaces
 
 RES = 512
 image = Image.open("sim/assets/langtable_prompt/frame_06.png")
@@ -44,6 +45,7 @@ def model(direction: str, genie=genie):
     return Image.fromarray(next_image)
 
 # Gradio function to handle user input
+@spaces.GPU
 def handle_input(direction):
     print(f"User clicked: {direction}")
     new_image = model(direction)  # Get a new image from the model

requirements.txt

@@ -13,6 +13,7 @@ diffusers  # used for stable diffusion tokenizers
 opencv-python
 tensorflow_datasets
 tensorflow
+spaces
 gcsfs
 
 # official MaskGIT