Update to use ZeroGPU spaces feature

app.py

@@ -11,6 +11,8 @@ from diffusers import StableDiffusion3Pipeline
 from diffusers import StableDiffusionPipeline
 from diffusers import DiffusionPipeline
 
+import spaces
+
 #from transformers import pipeline
 import gradio as gr
 
@@ -79,6 +81,7 @@ def base64_to_pil(img_base64):
 #    return result_image
 
 # Generate function
+@spaces.GPU(duration=120) # Designed to be effect-free in non-ZeroGPU environments, ensuring compatibility across different setups.
 def generate(prompt, negative_prompt, steps, guidance, width, height):
     # Ensure width and height are multiples of 8 (required by Stable Diffusion)
     width = int(width) - (int(width) % 8)