Hugging Face's logo

Spaces:
nroggendorff
/
cats
Runtime error

App Files Community
nroggendorff commited on
Commit
a550dff
·
verified ·
1 Parent(s): a99ce7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from diffusers import DiffusionPipeline
4
  def generate_image(steps):
5
  pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats", use_safetensors=True)
6
  pipe = pipeline.to("cuda")
7
- image = pipe().images[0]
8
  return image
9
 
10
  with gr.Blocks() as demo:
 
4
  def generate_image(steps):
5
  pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats", use_safetensors=True)
6
  pipe = pipeline.to("cuda")
7
+ image = pipe(num_inference_steps=steps).images[0]
8
  return image
9
 
10
  with gr.Blocks() as demo: