Update app.py

app.py

@@ -1280,14 +1280,14 @@ def build_ui():
 
     with gr.Blocks(title="TB X-ray Assistant (TBNet + RADIO)", css=css) as demo:
 
-# ---------------------------
-# Welcome screen (shown first)
-# ---------------------------
-with gr.Column(visible=True) as welcome_screen:
-    gr.Markdown('<div class="title">Welcome — TB X-ray Assistant (HF Spaces)</div>')
-
-    gr.Markdown(
-        f"""
+        # ---------------------------
+        # Welcome screen (shown first)
+        # ---------------------------
+        with gr.Column(visible=True) as welcome_screen:
+            gr.Markdown('<div class="title">Welcome — TB X-ray Assistant (HF Spaces)</div>')
+
+            gr.Markdown(
+                f"""
 <div class="card">
   <div style="font-size:16px; font-weight:900; margin-bottom:8px;">What this Space does</div>
   <div style="opacity:0.92;">
@@ -1362,9 +1362,9 @@ with gr.Column(visible=True) as welcome_screen:
   Device policy: <b>{DEVICE}</b> (FORCE_CPU={FORCE_CPU})
 </div>
 """
-    )
+            )
 
-    continue_btn = gr.Button("Continue →", variant="primary")
+            continue_btn = gr.Button("Continue →", variant="primary")
 
         # ---------------------------
         # Main app UI (hidden initially)
@@ -1388,7 +1388,11 @@ with gr.Column(visible=True) as welcome_screen:
                     tb_weights = gr.Textbox(label="TBNet weights (.pt)", value=DEFAULT_TB_WEIGHTS)
                     lung_weights = gr.Textbox(label="Lung U-Net weights (.pt)", value=DEFAULT_LUNG_WEIGHTS)
 
-                    backbone = gr.Dropdown(choices=["efficientnet_b0"], value="efficientnet_b0", label="TBNet backbone")
+                    backbone = gr.Dropdown(
+                        choices=["efficientnet_b0"],
+                        value="efficientnet_b0",
+                        label="TBNet backbone"
+                    )
 
                     threshold = gr.Slider(
                         0.01, 0.99, value=TBNET_SCREEN_THR, step=0.01,