Update app.py

app.py

@@ -32,7 +32,7 @@ def show_images_save(x):
     return grid_im
 
 def generate():
-    x = torch.randn(1, 4, 64, 64).to(device)
+    x = torch.randn(2, 4, 64, 64).to(device)
     # Minimal sampling loop
     for i, t in tqdm(enumerate(scheduler.timesteps)):
         model_input = scheduler.scale_model_input(x, t)