Karumoon commited on
Commit
cccf69a
1 Parent(s): a4c4e62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import gradio as gr
2
  from diffusers import DiffusionPipeline
3
 
4
- pipeline = DiffusionPipeline.from_pretrained("anton-l/ddpm-ema-flowers-64")
 
5
 
6
  generator.to("cuda")
7
 
 
1
  import gradio as gr
2
  from diffusers import DiffusionPipeline
3
 
4
+ generator = DiffusionPipeline.from_pretrained("anton-l/ddpm-ema-flowers-64")
5
+ #generator = DiffusionPipeline.from_pretrained("google/ddpm-celebahq-256")
6
 
7
  generator.to("cuda")
8