Update app.py

app.py

@@ -197,6 +197,7 @@ model_management.load_models_gpu([
 
 print("Setup complete!")
 
+@spaces.GPU
 def generate_image(prompt, structure_image, style_image, depth_strength=15, canny_strength=30, style_strength=0.5, steps=28, progress=gr.Progress(track_tqdm=True)):
     """Main generation function that processes inputs and returns the path to the generated image."""
     timestamp = random.randint(10000, 99999)
@@ -368,7 +369,7 @@ def generate_image(prompt, structure_image, style_image, depth_strength=15, cann
             # Fall back to the expected path
             return os.path.join("output", output_filename)
 
-@spaces.GPU
+
 with gr.Blocks(css="footer {visibility: hidden}") as app:
     gr.Markdown("# 🎨 FLUX Zen Style Depth+Canny")
     gr.Markdown("Flux[dev] Redux + Flux[dev] Depth and XLabs Canny based on the space FLUX Style Shaping")