Update app.py

app.py

@@ -129,12 +129,12 @@ def process_scan(image):
 
     # Step C: Route to Specific Model
     if mri_prob > xray_prob:
-        modality_status = f"MRI Brain Scan"
+        modality_status = f"MRI Brain Scan)"
         diagnostic_results = predict_mri(image)
         # We only want top 1 for MRI based on your previous UI setup
         top_k = 1 
     else:
-        modality_status = f"Chest X-Ray"
+        modality_status = f"Chest X-Ray)"
         diagnostic_results = predict_xray(image)
         # We want top 2 for X-Ray based on your previous UI setup
         top_k = 2
@@ -144,18 +144,15 @@ def process_scan(image):
 # ==========================================
 # 5. Define the Unified Gradio Interface
 # ==========================================
-with gr.Blocks(title="Medical Scan Classification") as interface:
-    gr.Markdown("# 🩺 Medical Scan Classifier")
-    gr.Markdown("Upload **any** scan (MRI Brain Scan or Chest X-Ray). The system will automatically detect the modality and route it to the appropriate diagnostic model.")
-    
+with gr.Blocks(title="BTech Project") as interface:
     with gr.Row():
         with gr.Column():
-            scan_input = gr.Image(label="Upload Medical Scan")
-            analyze_button = gr.Button("Analyze Scan", variant="primary")
+            scan_input = gr.Image(label="Upload XRay or MRI Image")
+            analyze_button = gr.Button("Predict", variant="primary")
         
         with gr.Column():
-            modality_output = gr.Textbox(label="Detection Routing Status", interactive=False)
-            diagnostic_output = gr.Label(label="Predicted Conditions")
+            modality_output = gr.Textbox(label="Image Type", interactive=False)
+            diagnostic_output = gr.Label(label="Prediction")
             
     analyze_button.click(
         fn=process_scan,