Spaces:
Runtime error
Runtime error
Commit
·
96e80a0
1
Parent(s):
d869637
Update app.py
Browse files
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 |
|