Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
 
1
  import gradio as gr
2
  import torch
3
  from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
4
  from huggingface_hub import hf_hub_download
5
  from safetensors.torch import load_file
6
- import spaces
7
  from PIL import Image
8
 
9
  SAFETY_CHECKER = True
@@ -46,7 +46,7 @@ if SAFETY_CHECKER:
46
  return images, has_nsfw_concepts
47
 
48
  # Function
49
- @spaces.GPU(enable_queue=True)
50
  def generate_image(prompt, ckpt):
51
  global loaded
52
  print(prompt, ckpt)
 
1
+ import spaces
2
  import gradio as gr
3
  import torch
4
  from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
5
  from huggingface_hub import hf_hub_download
6
  from safetensors.torch import load_file
 
7
  from PIL import Image
8
 
9
  SAFETY_CHECKER = True
 
46
  return images, has_nsfw_concepts
47
 
48
  # Function
49
+ @spaces.GPU() # enable=Queue is halfly deprecated.
50
  def generate_image(prompt, ckpt):
51
  global loaded
52
  print(prompt, ckpt)