Fix Gradio 6.0: move css to launch(), remove scale from Markdown

app.py

@@ -361,7 +361,7 @@ if gallery_data:
         gallery_name_map[label] = key
         gallery_display_names.append(label)
 
-with gr.Blocks(title="UNIStainNet — Virtual IHC Staining", css=CUSTOM_CSS) as demo:
+with gr.Blocks(title="UNIStainNet — Virtual IHC Staining") as demo:
 
     # ── Header ──────────────────────────────────────────────────────
     gr.HTML("""
@@ -394,14 +394,12 @@ with gr.Blocks(title="UNIStainNet — Virtual IHC Staining", css=CUSTOM_CSS) as
                 Each example shows the H&E input, ground truth IHC, and all 4 generated stains from our unified model.
             </div>
             """)
-            with gr.Row():
-                gallery_dropdown = gr.Dropdown(
-                    choices=gallery_display_names,
-                    value=gallery_display_names[0] if gallery_display_names else None,
-                    label="Select Example",
-                    scale=3,
-                )
-                gallery_info_box = gr.Markdown(value="", scale=2)
+            gallery_dropdown = gr.Dropdown(
+                choices=gallery_display_names,
+                value=gallery_display_names[0] if gallery_display_names else None,
+                label="Select Example",
+            )
+            gallery_info_box = gr.Markdown(value="")
 
             gr.HTML('<p class="section-heading">Input & Ground Truth</p>')
             with gr.Row():
@@ -574,4 +572,4 @@ with gr.Blocks(title="UNIStainNet — Virtual IHC Staining", css=CUSTOM_CSS) as
     """)
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft())
+    demo.launch(theme=gr.themes.Soft(), css=CUSTOM_CSS)