Added spaces

app.py

@@ -4,6 +4,7 @@ from injection_utils import regiter_attention_editor_diffusers
 from bounded_attention import BoundedAttention
 from pytorch_lightning import seed_everything
 
+import spaces
 import gradio as gr
 import torch
 import numpy as np
@@ -61,6 +62,7 @@ def inference(
     return model(prompts, latents=start_code, guidance_scale=classifier_free_guidance_scale).images
 
 
+@spaces.GPU
 def generate(
     device,
     model,

requirements.txt

@@ -20,3 +20,4 @@ torchvision==0.15.2
 tqdm==4.66.1
 transformers==4.32.0
 xformers==0.0.21
+spaces==0.24.2