rishabh5752 commited on
Commit
96e80a0
·
1 Parent(s): d869637

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def predict(image):
22
  label = 'Malignant'
23
  confidence = prediction[0][0] # Confidence for malignant
24
 
25
- return {'label': label, 'confidence': confidence}
26
 
27
  examples = [["benign.jpg"], ["malignant.jpg"]]
28
 
 
22
  label = 'Malignant'
23
  confidence = prediction[0][0] # Confidence for malignant
24
 
25
+ return {'label': label, 'confidence': float(confidence)} # Convert confidence to float
26
 
27
  examples = [["benign.jpg"], ["malignant.jpg"]]
28