Update app.py

app.py

@@ -32,7 +32,7 @@ def show_images_save(x):
     return grid_im
 
 def generate():
-    x = torch.randn(2, 4, 64, 64).to(device)
+    x = torch.randn(1, 4, 64, 64).to(device)
     # Minimal sampling loop
     for i, t in tqdm(enumerate(scheduler.timesteps)):
         model_input = scheduler.scale_model_input(x, t)
@@ -43,7 +43,7 @@ def generate():
     return show_images_save(x).resize((64, 64), resample=Image.NEAREST)
     
 
-outputs = gr.Image(label="result")
+outputs = (gr.Image(label="result"), gr.Image(label="result"))
 
 # Setting up a minimal interface to our function:
 demo = gr.Interface(