Update app.py

app.py

@@ -27,7 +27,6 @@ def load_model():
 @st.cache_resource
 def load_mlb():
     # Define the classes based on your label set
-    # classes = ['E11.9', 'I10', 'J45.909', 'M54.5', 'N39.0', '81001.0', '99213.0', '99214.0']
     classes = ['81001.0','99213.0','99214.0','E11.9','I10','J45.909','M54.5','N39.0']
     # Initialize and fit the MultiLabelBinarizer
     mlb = MultiLabelBinarizer(classes=classes)
@@ -51,10 +50,10 @@ mlb = load_mlb()
 
 # Streamlit UI
 st.title("Automated Medical Coding Prediction")
-# st.write("Enter clinical notes to predict CPT and ICD codes.")
+# st.write("Enter clinical notes to predict ICD and CPT codes.")
 
 # Text input for Clinical Notes
-clinical_note = st.text_area("Enter clinical notes to predict ICD and CPT codes")
+clinical_note = st.text_area("Enter clinical notes")
 
 # Prediction button
 if st.button('Predict'):