Spaces:

1aurent
/

ddpm-mnist

Sleeping

stevhliu HF staff commited on Mar 7, 2023

Commit

532834a

1 Parent(s): b84ee6b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ pipeline = DiffusionPipeline.from_pretrained("johnowhitaker/ddpm-butterflies-32p
 def predict(steps, seed):
     generator = torch.manual_seed(seed)
     for i in range(1,steps):
-        yield pipeline(generator=generator, num_inference_steps=i)["sample"][0]
 random_seed = random.randint(0, 2147483647)
 gr.Interface(

 def predict(steps, seed):
     generator = torch.manual_seed(seed)
     for i in range(1,steps):
+        yield pipeline(generator=generator, num_inference_steps=i).images[0]
 random_seed = random.randint(0, 2147483647)
 gr.Interface(