Update app.py

app.py

@@ -37,9 +37,9 @@ def predict_bmwX(image):
     prediction_dict = {class_names[i]: np.round(float(prediction[0][i]), 2) for i in range(len(class_names))}
     
     # Sort the dictionary by value in descending order and get the top 3 classes
-    top_3 = dict(sorted(prediction_dict.items(), key=lambda item: item[1], reverse=True)[:3])
+    sorted_predictions = dict(sorted(prediction_dict.items(), key=lambda item: item[1], reverse=True))
         
-    return top_3
+    return sorted_predictions
 
 
 input_image = gr.Image()