ZeroGPU

app.py

@@ -1,4 +1,5 @@
 import torch
+import spaces
 import gradio as gr
 from diffusers import DiffusionPipeline
 
@@ -12,7 +13,7 @@ pipe = DiffusionPipeline.from_pretrained(
 pipe.to("cuda")
 print("Pipeline loaded!")
 
-
+@spaces.GPU
 def generate_image(prompt, height, width, num_inference_steps, seed, randomize_seed):
     """Generate an image from the given prompt."""
     if randomize_seed: