Spaces:
Runtime error
Runtime error
Commit
·
3ef24f4
1
Parent(s):
4940ea7
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,10 @@ def predict(image):
|
|
| 20 |
else:
|
| 21 |
label = 'Malignant'
|
| 22 |
|
| 23 |
-
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
examples = [["benign.jpg"], ["malignant.jpg"]]
|
| 26 |
|
|
|
|
| 20 |
else:
|
| 21 |
label = 'Malignant'
|
| 22 |
|
| 23 |
+
# Get the sigmoid percentage
|
| 24 |
+
sigmoid_percentage = f"{prediction[0][0] * 100:.2f}%"
|
| 25 |
+
|
| 26 |
+
return f"{label} ({sigmoid_percentage})", sigmoid_percentage
|
| 27 |
|
| 28 |
examples = [["benign.jpg"], ["malignant.jpg"]]
|
| 29 |
|