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

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
  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
 
9
 
10
  def greet(name):
 
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
 
9
 
10
  def greet(name):