rafaaa2105 commited on
Commit
5bde01d
1 Parent(s): 3871549

Add ZERO decorations

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from diffusers.models import AutoencoderKL
4
  from diffusers import StableDiffusionXLPipeline
5
  import gradio as gr
6
  import requests
 
7
 
8
  models_list = []
9
  loras_list = ["None"]
@@ -82,6 +83,7 @@ def load_model(model, lora="", use_lora=False):
82
  except Exception as e:
83
  return f"Error loading model {model}: {e}"
84
 
 
85
  def generate_images(
86
  model_name,
87
  lora_name,
 
4
  from diffusers import StableDiffusionXLPipeline
5
  import gradio as gr
6
  import requests
7
+ import spaces
8
 
9
  models_list = []
10
  loras_list = ["None"]
 
83
  except Exception as e:
84
  return f"Error loading model {model}: {e}"
85
 
86
+ @spaces.GPU
87
  def generate_images(
88
  model_name,
89
  lora_name,