rishabh5752 commited on
Commit
3ef24f4
·
1 Parent(s): 4940ea7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -20,7 +20,10 @@ def predict(image):
20
  else:
21
  label = 'Malignant'
22
 
23
- return label
 
 
 
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