Update app.py

app.py

@@ -152,9 +152,9 @@ if __name__ == "__main__":
                 seed = gr.Slider(minimum=0, maximum=1000000, step=1, label="seed", value=0)
                 timeshift = gr.Slider(minimum=0.1, maximum=5.0, step=0.1, label="timeshift", value=3.0)
                 order = gr.Slider(minimum=1, maximum=4, step=1, label="order", value=2)
-            with gr.Column(scale=2):
+            with gr.Column(scale=1):
                 btn = gr.Button("Generate")
-                output_sample = gr.Gallery(label="Images")
+                output_sample = gr.Image(label="Images")
             # with gr.Column(scale=2):
             #     output_trajs = gr.Gallery(label="Trajs of Diffusion", columns=2, rows=2)