Update app.py

app.py

@@ -26,7 +26,14 @@ if st.button("Submit"):
     # Get prediction
     prediction = clf.predict(X)[0]
     
-    st.text("Predicted Disease: " + prediction)
+    st.write("Predicted Disease: " + prediction)
+
+    # You can also display the top N predicted classes with their probabilities
+    top_n = 5  # Change this value to display more or fewer top classes
+    top_n_classes = np.argsort(-confidence_scores)[:top_n]
+    st.text("Top {} Predicted Diseases:".format(top_n))
+    for i, class_idx in enumerate(top_n_classes):
+        st.text(f"{clf.classes_[class_idx]}: {confidence_scores[class_idx]:.2f}")