update

app.py

@@ -109,7 +109,8 @@ def interface():
         )
 
         # Model Selection and Setup
-        with gr.Box(elem_id="container"):
+        with gr.Column(elem_id="container"):
+            gr.Markdown("<h2 style='color: #6A1B9A; text-align: center;'>Model Selection and Setup</h2>")
             with gr.Row():
                 model_dropdown = gr.Dropdown(
                     label="Select Model",
@@ -125,8 +126,8 @@ def interface():
                 setup_status = gr.Textbox(label="Setup Status", interactive=False, placeholder="Setup status will appear here...")
                 
         # Inference Section
-        with gr.Box(elem_id="container"):
-            gr.Markdown("<h2>BitNet Inference</h2>")
+        with gr.Column(elem_id="container"):
+            gr.Markdown("<h2 style='color: #6A1B9A; text-align: center;'>BitNet Inference</h2>")
             with gr.Row():
                 num_tokens = gr.Slider(
                     minimum=1, maximum=100, 
@@ -135,7 +136,8 @@ def interface():
                 )
                 input_text = gr.Textbox(
                     label="Input Text", 
-                    placeholder="Enter your input text here..."
+                    placeholder="Enter your input text here...",
+                    value="Who is Zeus?"
                 )
             with gr.Row():
                 infer_button = gr.Button("Run Inference")
@@ -143,8 +145,8 @@ def interface():
                 time_output = gr.Textbox(label="Inference Time", interactive=False, placeholder="Inference time will appear here...")
 
         # Comparison with Transformers Section
-        with gr.Box(elem_id="container"):
-            gr.Markdown("<h2>Compare with Transformers</h2>")
+        with gr.Column(elem_id="container"):
+            gr.Markdown("<h2 style='color: #6A1B9A; text-align: center;'>Compare with Transformers</h2>")
             with gr.Row():
                 transformer_model_dropdown = gr.Dropdown(
                     label="Select Transformers Model",
@@ -152,7 +154,7 @@ def interface():
                     value="TinyLlama/TinyLlama_v1.1",
                     interactive=True
                 )
-                input_text_tr = gr.Textbox(label="Input Text", placeholder="Enter your input text here...")
+                input_text_tr = gr.Textbox(label="Input Text", placeholder="Enter your input text here...", value="Who is Zeus?")
             with gr.Row():
                 compare_button = gr.Button("Run Transformers Inference")
                 transformer_result_output = gr.Textbox(label="Transformers Output", interactive=False, placeholder="Transformers output will appear here...")