Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -129,12 +129,12 @@ def process_scan(image):
|
|
| 129 |
|
| 130 |
# Step C: Route to Specific Model
|
| 131 |
if mri_prob > xray_prob:
|
| 132 |
-
modality_status = f"MRI Brain Scan
|
| 133 |
diagnostic_results = predict_mri(image)
|
| 134 |
# We only want top 1 for MRI based on your previous UI setup
|
| 135 |
top_k = 1
|
| 136 |
else:
|
| 137 |
-
modality_status = f"Chest X-Ray
|
| 138 |
diagnostic_results = predict_xray(image)
|
| 139 |
# We want top 2 for X-Ray based on your previous UI setup
|
| 140 |
top_k = 2
|
|
|
|
| 129 |
|
| 130 |
# Step C: Route to Specific Model
|
| 131 |
if mri_prob > xray_prob:
|
| 132 |
+
modality_status = f"MRI Brain Scan"
|
| 133 |
diagnostic_results = predict_mri(image)
|
| 134 |
# We only want top 1 for MRI based on your previous UI setup
|
| 135 |
top_k = 1
|
| 136 |
else:
|
| 137 |
+
modality_status = f"Chest X-Ray"
|
| 138 |
diagnostic_results = predict_xray(image)
|
| 139 |
# We want top 2 for X-Ray based on your previous UI setup
|
| 140 |
top_k = 2
|