updated code

app.py

@@ -59,6 +59,7 @@ def view_model(selected_models):
             """
     return netron_html if netron_html else "<p>No valid models selected for visualization.</p>"
 
+# CSS to style the Gradio components and HTML content
 custom_css = """
 .custom-button {
     background-color: purple !important;
@@ -66,6 +67,10 @@ custom_css = """
     width: 120px !important;
     border-radius: 5px !important;
     font-size: 14px !important;
+    display: inline-block !important;
+    padding: 10px 20px !important;
+    text-align: center !important;
+    cursor: pointer !important;
 }
 
 .custom-row {
@@ -92,8 +97,6 @@ with gr.Blocks(css=custom_css) as interface:
             <p>With the integration of state-of-the-art YOLO models, you can explore the performance of object detection algorithms on various images.</p>
         </div>
     """)
-
-    
     
     # Default sample
     default_sample = "Sample 1"
@@ -134,7 +137,6 @@ with gr.Blocks(css=custom_css) as interface:
             height="auto",       # Adjust height automatically based on content
             columns=1 ,
             object_fit="contain"
-
         ) 
         netron_display = gr.HTML(label="Netron Visualization")
 
@@ -144,6 +146,7 @@ with gr.Blocks(css=custom_css) as interface:
         inputs=sample_selection,
         outputs=sample_display,
     )
+
     with gr.Row(elem_classes="custom-row"):
         dff_gallery = gr.Gallery(
             label="Deep Feature Factorization",
@@ -153,7 +156,6 @@ with gr.Blocks(css=custom_css) as interface:
             height="auto"    # Adjust as needed
         ) 
 
-
     # Multi-threaded processing
     def run_both(sample_choice, uploaded_image, selected_models):
         results = []
@@ -190,7 +192,6 @@ with gr.Blocks(css=custom_css) as interface:
         return [(image1, text)], netron_html, image2
 
     # Run button click
-
     run_button.click(
         fn=run_both,
         inputs=[sample_selection, upload_image, selected_models],