ZeroGPU: must import before any CUDA-related packages

app.py

@@ -1,6 +1,13 @@
 import colorsys
 import os
 
+# ZeroGPU: must import before any CUDA-related packages
+try:
+    import spaces
+    GPU_DECORATOR = spaces.GPU
+except ImportError:
+    GPU_DECORATOR = lambda func: func
+
 import gradio as gr
 import matplotlib.colors as mcolors
 import numpy as np
@@ -8,13 +15,6 @@ import torch
 from PIL import Image, ImageDraw, ImageFont
 from transformers import SegformerForSemanticSegmentation, SegformerImageProcessor
 
-# Handle spaces.GPU decorator for local vs HuggingFace execution
-try:
-    import spaces
-    GPU_DECORATOR = spaces.GPU
-except ImportError:
-    GPU_DECORATOR = lambda func: func
-
 # ----------------- CONFIG ----------------- #
 
 ASSETS_DIR = os.path.join(os.path.dirname(__file__), "assets")