bauckluc commited on
Commit
1985e03
1 Parent(s): ee580df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -48,7 +48,6 @@ def predict_breed(image):
48
  # Create a dictionary with the probabilities for each dog breed
49
  prediction_dict = {class_names[i]: np.round(float(prediction[0][i]), 2) for i in range(len(class_names))}
50
 
51
- # Sort the dictionary by value in descending order and get the top 3 classes
52
  sorted_predictions = dict(sorted(prediction_dict.items(), key=lambda item: item[1], reverse=True))
53
 
54
  return sorted_predictions
 
48
  # Create a dictionary with the probabilities for each dog breed
49
  prediction_dict = {class_names[i]: np.round(float(prediction[0][i]), 2) for i in range(len(class_names))}
50
 
 
51
  sorted_predictions = dict(sorted(prediction_dict.items(), key=lambda item: item[1], reverse=True))
52
 
53
  return sorted_predictions