Update app.py

changed the number of sample images

app.py

@@ -172,7 +172,7 @@ def p_sample_loop(model, shape):
     img = torch.randn(shape, device=device)
     imgs = []
 
-    for i in tqdm(reversed(range(0, timesteps)), desc='sampling loop time step', total=timesteps):
+    for i in tqdm(reversed(range(0, timesteps)), desc='sampling loop time step', total=10):
         img = p_sample(model, img, torch.full((b,), i, device=device, dtype=torch.long), 3)
         imgs.append(img.cpu().numpy())
     return imgs