Hugging Face's logo

Spaces:
nroggendorff
/
cats
Paused

App Files Community
nroggendorff commited on
Commit
a9cbedb
·
verified ·
1 Parent(s): e9f9fa4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
  from diffusers import DiffusionPipeline
3
 
4
- def generate_image(seed):
5
  pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
6
  pipe = pipeline.to("cuda")
7
- image = pipe(seed=int(seed)).images[0]
8
  return image
9
 
10
  with gr.Blocks() as demo:
 
1
  import gradio as gr
2
  from diffusers import DiffusionPipeline
3
 
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: