Port to ZeroGPU: import spaces, decorate segment with @spaces.GPU

app.py

@@ -14,6 +14,8 @@ used for offline testing.
 
 import os
 import threading
+
+import spaces  # ZeroGPU: must be imported before torch so CUDA emulation is active at load time
 import numpy as np
 import torch
 from PIL import Image
@@ -113,6 +115,7 @@ def _legend_html(pairs, cfg):
 _EMPTY_LEGEND = "<div class='legend'><span class='muted'>Run a segmentation to see detected classes.</span></div>"
 
 
+@spaces.GPU(duration=120)
 @torch.no_grad()
 def segment(image, dataset=DEFAULT_DATASET, alpha=0.55):
     if image is None: