Zero-GPU

app.py

@@ -7,6 +7,7 @@ import random
 import gradio as gr
 import numpy as np
 import uuid
+import spaces
 from diffusers import ConsistencyDecoderVAE, DPMSolverMultistepScheduler, Transformer2DModel, AutoencoderKL, SASolverScheduler
 import torch
 from typing import Tuple
@@ -191,6 +192,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
 @torch.no_grad()
 @torch.inference_mode()
+@spaces.GPU(duration=120)
 def generate(
         prompt: str,
         negative_prompt: str = "",