Update app.py
Browse files
app.py
CHANGED
@@ -79,7 +79,7 @@ def predict_image(input):
|
|
79 |
predicted_class_label = class_labels[predicted_class_index]
|
80 |
|
81 |
# Get the confidence score of the predicted class
|
82 |
-
confidence_score =
|
83 |
|
84 |
# Return predicted class label and confidence score
|
85 |
return {predicted_class_label: confidence_score}
|
|
|
79 |
predicted_class_label = class_labels[predicted_class_index]
|
80 |
|
81 |
# Get the confidence score of the predicted class
|
82 |
+
confidence_score = predictions[0][predicted_class_index]
|
83 |
|
84 |
# Return predicted class label and confidence score
|
85 |
return {predicted_class_label: confidence_score}
|