fffiloni commited on
Commit
9bb4f06
·
1 Parent(s): 9c5cafb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import gradio as gr
2
- import spaces
3
  from diffusers import StableDiffusionXLPipeline, UNet2DConditionModel
4
  import torch
5
 
@@ -16,7 +15,6 @@ pipe = pipe.to("cuda")
16
  pipe.enable_model_cpu_offload()
17
  pipe.enable_vae_slicing()
18
 
19
- @spaces.GPU
20
  def infer(prompt):
21
  image = pipe(prompt, guidance_scale=7.5).images[0]
22
  return image
 
1
  import gradio as gr
 
2
  from diffusers import StableDiffusionXLPipeline, UNet2DConditionModel
3
  import torch
4
 
 
15
  pipe.enable_model_cpu_offload()
16
  pipe.enable_vae_slicing()
17
 
 
18
  def infer(prompt):
19
  image = pipe(prompt, guidance_scale=7.5).images[0]
20
  return image