Update app.py

app.py

@@ -13,6 +13,7 @@ from kolors.models.unet_2d_condition import UNet2DConditionModel
 from diffusers import AutoencoderKL, EulerDiscreteScheduler
 
 from huggingface_hub import snapshot_download
+import spaces 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
@@ -58,6 +59,7 @@ pipe.load_ip_adapter(f'{root_dir}/weights/Kolors-IP-Adapter-Plus', subfolder="",
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
+@spaces.GPU
 def infer(prompt, ip_adapter_image, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=50, ip_adapter_scale=0.5, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)