CGAllenger commited on
Commit
dc5ceea
·
verified ·
1 Parent(s): efc5d81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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