Update app.py

app.py

@@ -25,7 +25,7 @@ class_indices = load_labels()
 class_labels = list(class_indices.keys())
 
 # Title
-st.title(" Animal Image Classifier")
+st.title("Animal Image Classifier")
 st.markdown("Upload up to **3 images** to get predictions")
 
 # File uploader
@@ -55,18 +55,13 @@ if uploaded_files:
 
                 # Prediction
                 preds = model.predict(img_array)[0]
-                top_indices = preds.argsort()[-3:][::-1]
-                top_labels = [class_labels[i] for i in top_indices]
-                top_scores = [preds[i] for i in top_indices]
+                top_index = np.argmax(preds)
+                top_label = class_labels[top_index]
 
-                # Display predictions
-                st.success(f"Top Prediction: **{top_labels[0].capitalize()}**")
-                st.markdown("##### Top 3 Predictions:")
-                for label, score in zip(top_labels, top_scores):
-                    st.write(f"**{label.capitalize()}** — {score:.4f}")
+                # Display top prediction only
+                st.success(f"Top Prediction: **{top_label.capitalize()}**")
 
             except UnidentifiedImageError:
                 st.error("Error: Invalid image file.")
             except Exception as e:
                 st.error(f"Unexpected error: {str(e)}")
-