test

app.py

@@ -13,7 +13,6 @@ CONTROLNET_FILENAME = "ControlAny-SDXL/anime_canny/controlnet.safetensors"
 CACHE_DIR = None
 
 
-@spaces.GPU
 def ui():
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model_file = hf_hub_download(
@@ -31,7 +30,8 @@ def ui():
         filename=CONTROLNET_FILENAME,
         cache_dir=CACHE_DIR,
     )
-    pipeline = tools.get_pipeline(
+
+    pipeline = spaces.GPU(tools.get_pipeline)(
         pretrained_model_name_or_path=model_file,
         unet_model_name_or_path=unet_file,
         controlnet_model_name_or_path=controlnet_file,
@@ -109,6 +109,7 @@ def ui():
                     show_share_button=True,
                 )
 
+        @spaces.GPU
         def generate(
             prompt,
             control_image,