Update app.py

app.py

@@ -156,7 +156,7 @@ if __name__ == "__main__":
                 btn = gr.Button("Generate")
                 output_sample = gr.Image(label="Images")
             with gr.Column(scale=1):
-                 output_trajs = gr.Image(label="Trajs of Diffusion", columns=2, rows=2)
+                 output_trajs = gr.Image(label="Trajs of Diffusion")
 
         btn.click(fn=pipeline,
                   inputs=[