Spaces:

1aurent
/

ddpm-mnist

Running on Zero

stevhliu HF Staff commited on Mar 7, 2023

Commit

210eb75

1 Parent(s): 532834a

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 = DiffusionPipeline.from_pretrained("johnowhitaker/ddpm-butterflies-32px")
 def predict(steps, seed):
     generator = torch.manual_seed(seed)

 import random
 import numpy as np
+pipeline = DiffusionPipeline.from_pretrained("google/ddpm-ema-cat-256")
 def predict(steps, seed):
     generator = torch.manual_seed(seed)