Jamshaid89 commited on
Commit
6aba54b
·
1 Parent(s): 9deaebf
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -96,7 +96,7 @@ def predict_xray(image: np.ndarray):
96
 
97
  # Generate results with probabilities and binary labels (threshold=0.5)
98
  result = {
99
- CLASS_NAMES[str(i)]: f"{probs[i]:.3f} ({'Positive' if probs[i] >= 0.5 else 'Negative'})"
100
  for i in range(len(CLASS_NAMES)) if i != 10 # Exclude "No Finding"
101
  }
102
 
 
96
 
97
  # Generate results with probabilities and binary labels (threshold=0.5)
98
  result = {
99
+ CLASS_NAMES[str(i)]: probs[i]
100
  for i in range(len(CLASS_NAMES)) if i != 10 # Exclude "No Finding"
101
  }
102