Update app.py

app.py

@@ -48,7 +48,11 @@ def classify_image(input_image):
 
     # Perform prediction
     prediction = inception_net.predict(input_image).flatten()
-    confidences = {labels[i]: float(prediction[i]) for i in range(len(labels))}
+
+    # Get the top indices and their respective confidence scores
+    top_indices = np.argsort(prediction)[-3:][::-1]  # Get the top 3 indices
+    confidences = {labels[i]: float(prediction[i]) for i in top_indices}
+    
     return confidences
 
 image = gr.Image(interactive=True, label="Upload Image")