Spaces:
Sleeping
Sleeping
Commit ·
6aba54b
1
Parent(s): 9deaebf
dsf
Browse files
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)]:
|
| 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 |
|