UX4567 commited on
Commit
1b2e644
·
verified ·
1 Parent(s): 99fdab6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
22
  ).to(device)
23
 
24
  # 3. Generation function with GPU decorator
25
- @spaces.gpu
26
  def generate_image(prompt):
27
  image = pipe(prompt, num_inference_steps=35, guidance_scale=7.5).images[0]
28
  return image
 
22
  ).to(device)
23
 
24
  # 3. Generation function with GPU decorator
25
+ @spaces.GPU
26
  def generate_image(prompt):
27
  image = pipe(prompt, num_inference_steps=35, guidance_scale=7.5).images[0]
28
  return image