Add ZeroGPU support with spaces decorator

app.py

@@ -37,12 +37,22 @@ except ImportError:
     PLOTLY_AVAILABLE = False
     print("Plotly not available, charts will be disabled")
 
+# Try to import spaces for ZeroGPU support
+try:
+    from spaces import GPU
+    SPACES_AVAILABLE = True
+except ImportError:
+    SPACES_AVAILABLE = False
+    print("HuggingFace spaces not available (running locally)")
+
 # ============================================================================
 # Configuration
 # ============================================================================
 
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 print(f"Device: {DEVICE}")
+if SPACES_AVAILABLE:
+    print("ZeroGPU support enabled")
 
 # Check CUDA kernels availability
 try:
@@ -961,7 +971,7 @@ def create_benchmark_comparison(style: str) -> str:
 # Gradio Interface Functions
 # ============================================================================
 
-def stylize_image(
+def stylize_image_impl(
     input_image: Optional[Image.Image],
     style: str,
     backend: str,
@@ -1077,6 +1087,13 @@ def stylize_image(
         return None, error_msg, None
 
 
+# Wrap with GPU decorator for ZeroGPU if available
+if SPACES_AVAILABLE:
+    stylize_image = GPU(stylize_image_impl)
+else:
+    stylize_image = stylize_image_impl
+
+
 def process_webcam_frame(image: Image.Image, style: str, backend: str) -> Image.Image:
     """Process webcam frame in real-time."""
     if image is None:

requirements.txt

@@ -3,6 +3,7 @@ torch>=2.0.0
 torchvision>=0.15.0
 gradio==4.32.0
 huggingface_hub==0.20.0
+spaces
 Pillow>=9.5.0
 numpy>=1.24.0