Add ZeroGPU decorator for H200

app.py

@@ -7,6 +7,7 @@ import gradio as gr
 import torch
 import torch.nn as nn
 from pathlib import Path
+import spaces
 
 # Machine limits
 BOUNDS = {"left": -420.5, "right": 420.5, "top": 594.5, "bottom": -594.5}
@@ -292,6 +293,7 @@ def gcode_to_svg(gcode: str) -> str:
     return svg
 
 
+@spaces.GPU
 def generate(prompt: str, temperature: float, max_tokens: int, num_steps: int, guidance: float):
     """Generate gcode from text prompt via SD-Gcode diffusion."""
     if not prompt or not prompt.strip():