Spaces:

1aurent
/

ddpm-mnist

Sleeping

stevhliu HF Staff commited on Mar 7, 2023

Commit

85398ff

1 Parent(s): ee677a6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import gradio as gr
 import random
 import numpy as np
-pipeline = DDPMPipeline.from_pretrained("stevhliu/ddpm-butterflies-128")
 def predict(steps, seed):
     generator = torch.manual_seed(seed)

 import random
 import numpy as np
+pipeline = DDPMPipeline.from_pretrained("johnowhitaker/ddpm-butterflies-32px")
 def predict(steps, seed):
     generator = torch.manual_seed(seed)