rocioadlc commited on
Commit
9d149e4
Β·
verified Β·
1 Parent(s): cb3043d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = (predictions, -1)[predicted_class_index]
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}