change layout

app.py

@@ -69,8 +69,8 @@ TITLE_IMAGE = """
 """
 
 TITLE = """
-<div style="display: inline-flex; align-items: center; text-align: center; max-width: 1400px; gap: 0.8rem; font-size: 2.2rem;">
-<h1 style="font-weight: 900; margin-bottom: 10px; margin-top: 10px;">
+<div style="text-align: center; max-width: 1400px; margin: 0 auto;">
+<h1 style="font-weight: 900; margin-bottom: 10px; margin-top: 10px; font-size: 2.2rem;">
     🤗 Optimum Neuron Model Exporter (WIP)
 </h1>
 </div>
@@ -78,11 +78,13 @@ TITLE = """
 
 DESCRIPTION = """
 Export 🤗 Transformers models hosted on the Hugging Face Hub to AWS Neuron-optimized format for Inferentia/Trainium acceleration.
-*Features:*
+
+**Features:**
 - Automatically opens PR with Neuron-optimized model
 - Preserves original model weights
 - Adds proper tags to model card
-*Note:*
+
+**Note:**
 - PR creation requires the Space owner to have a valid write token set via HF_WRITE_TOKEN
 """
 
@@ -143,26 +145,31 @@ CUSTOM_CSS = """
 """
 
 with gr.Blocks(css=CUSTOM_CSS) as demo:
+    # Centered title and image
     gr.HTML(TITLE_IMAGE)
     gr.HTML(TITLE)
-
+    
+    # Full-width description
+    gr.Markdown(DESCRIPTION)
+    
+    # Input controls in a row at the bottom
     with gr.Row():
-        with gr.Column(scale=50):
-            gr.Markdown(DESCRIPTION)
-
-        with gr.Column(scale=50):
-            input_model = HuggingfaceHubSearch(
-                label="Hub model ID",
-                placeholder="Search for model ID on the hub",
-                search_type="model",
-            )
-            input_task = gr.Textbox(
-                value="auto",
-                max_lines=1,
-                label='Task (can be left to "auto", will be automatically inferred)',
-            )
-            btn = gr.Button("Export to Neuron")
-            output = gr.Markdown(label="Output")
+        input_model = HuggingfaceHubSearch(
+            label="Hub model ID",
+            placeholder="Search for model ID on the hub",
+            search_type="model",
+        )
+        input_task = gr.Textbox(
+            value="auto",
+            max_lines=1,
+            label='Task (can be left to "auto", will be automatically inferred)',
+        )
+    
+    # Export button below the inputs
+    btn = gr.Button("Export to Neuron", size="lg")
+    
+    # Output section
+    output = gr.Markdown(label="Output")
 
     btn.click(
         fn=neuron_export,