Fix HF ZeroGPU support: Add spaces decorator

requirements.txt

@@ -11,3 +11,4 @@ tensorflow-text; sys_platform != 'win32'
 tensorflow
 pydicom
 gradio
+spaces

src/app.py

@@ -46,6 +46,17 @@ def load_models():
             logger.error(f"Failed to load models: {e}")
             raise e
 
+# ZeroGPU compatibility for Hugging Face Spaces
+try:
+    import spaces
+except ImportError:
+    # Dummy decorator if running locally without spaces installed
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            return func
+
+@spaces.GPU
 def predict(image):
     if image is None:
         return "No image uploaded.", 0.0, "Please upload an image."