Spaces:

B-Bhuvanesh
/

diffusion

Sleeping

B-Bhuvanesh commited on Apr 16

Commit

61535ba

verified ·

1 Parent(s): 3dbefff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,7 @@ import gradio as gr
 from diffusers import DDPMPipeline
 # Load model
-pipe = DDPMPipeline.from_pretrained("google/ddpm-mnist")
 def generate():
     image = pipe().images[0]
     return image

 from diffusers import DDPMPipeline
 # Load model
+pipe = DDPMPipeline.from_pretrained("google/ddpm-cifar10-32")
 def generate():
     image = pipe().images[0]
     return image