Update app.py

app.py

@@ -17,9 +17,6 @@ for symptom in options:
 # Create a DataFrame from the dictionary
 X = pd.DataFrame([symptoms_dict], columns=column_list)
 
-st.write('Generated DataFrame:')
-st.write(X)
-
 if st.button("Submit"):
     # Unpickle classifier
     clf = joblib.load("self_diagnose_model.pkl")
@@ -30,13 +27,6 @@ if st.button("Submit"):
     
     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}")
-