Update app.py

app.py

@@ -17,6 +17,7 @@ from matplotlib.animation import FuncAnimation, PillowWriter
 from mpl_toolkits.mplot3d import Axes3D
 from mpl_toolkits.mplot3d.art3d import Poly3DCollection
 from visualization.plot_3d_global import draw_to_batch
+import imageio
 import sys
 import os
 import warnings
@@ -96,18 +97,18 @@ def infer(text):
         return "Error generating GIF. Please try again."
 
 with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") as demo:
-    with gr.Column():
+    with gr.Column(elem_id="col-container"):
         gr.Markdown("## 3D Human Motion Generation")
-        text_input = gr.Textbox(label="Enter the human action to generate", placeholder="Enter text description for the action here...")
-        output_image = gr.Image(label="Generated Human Motion", type="filepath")
-        submit_button = gr.Button("Generate Motion")
-
-        submit_button.click(
-            fn=infer,
-            inputs=[text_input],
-            outputs=[output_image]
-        )
-
+        with gr.Row():
+            text_input = gr.Textbox(label="Enter the human action to generate", placeholder="Enter text description for the action here...", show_label=False, max_lines=1, container=False)
+            submit_button = gr.Button("Generate Motion")
+        output_image = gr.Image(label="Generated Human Motion", type="filepath", show_label=False)
+
+    submit_button.click(
+        fn=infer,
+        inputs=[text_input],
+        outputs=[output_image]
+    )
 
 if __name__ == "__main__":
     demo.launch()