ZeroGPU: declare GPU function + space requirements

app.py

@@ -4,6 +4,7 @@ import os
 import gradio as gr
 import numpy as np
 import onnxruntime as ort
+import spaces
 from PIL import Image
 
 from inference_utils import build_grid, pil_to_np, resize_pad
@@ -15,6 +16,7 @@ IMAGE_SIZE = 512
 _sess = ort.InferenceSession(MODEL_PATH, providers=["CPUExecutionProvider"])
 
 
+@spaces.GPU(duration=60)
 def decompose(image: Image.Image) -> Image.Image:
     img_rgb = resize_pad(image.convert("RGB"), IMAGE_SIZE)
     x = pil_to_np(img_rgb).astype(np.float32)