Update demo.py

demo.py

@@ -52,6 +52,7 @@ css = """
     width: 250px;  /* Set all elements to the same width */
     margin: 5px;
 }
+
 /* Adjusting layout for Input Text and Inference Result */
 .input-row {
     display: flex;
@@ -66,8 +67,11 @@ css = """
 .output-text {
     flex: 1;  /* 1 part of the row */
 }
+
+/* Set button widths to match the Select Model width */
 .gradio-row button {
-    width: 500px;
+    width: 250px;  /* Same as the select box width */
+    margin: 5px;
 }
 """
 
@@ -90,7 +94,7 @@ with gr.Blocks(css=css) as app:
             label="Input Text",
             placeholder="Enter Text Here",
             lines=8,
-            elem_classes=["input-text"],
+            elem_classes=["input-text"],  # Applying the CSS class
         )
         output = gr.Textbox(
             label="Inference Result",
@@ -111,7 +115,6 @@ with gr.Blocks(css=css) as app:
         submit_button = gr.Button("Run Detection", variant="primary")
         clear_button = gr.Button("Clear", variant="secondary")
     
-    # Hooking up the change_mode function to select different models
     submit_button.click(run_test_power, inputs=[model_name, input_text, input_text], outputs=output)
     clear_button.click(lambda: ("", ""), inputs=[], outputs=[input_text, output])
 
@@ -133,4 +136,4 @@ with gr.Blocks(css=css) as app:
         ```
         """)
 
-app.launch()
+app.launch()