Hugging Face's logo

Spaces:
nroggendorff
/
cats
Runtime error

App Files Community
nroggendorff commited on
Commit
772093a
·
verified ·
1 Parent(s): 6a87bc6

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():
5
  pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
6
  pipe = pipeline.to("cuda")
7
- image = pipe(sample_size=512).images[0]
8
  return image
9
 
10
  with gr.Blocks() as demo:
 
4
  def generate_image():
5
  pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
6
  pipe = pipeline.to("cuda")
7
+ image = pipe().images[0]
8
  return image
9
 
10
  with gr.Blocks() as demo: