Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|