Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import joblib
 import matplotlib.pyplot as plt
 import time
+import base 
 
 # Load the pre-trained numerical imputer, scaler, and model using joblib
 num_imputer = joblib.load('numerical_imputer.joblib')
@@ -101,6 +102,12 @@ def main():
             st.markdown(f"Prediction: **{prediction_text}**")
             st.markdown(f"{status_icon} {sepsis_explanation}")
             st.write(output_df)
+
+            # Add a download button for output_df
+            csv = output_df.to_csv(index=False)
+            b64 = base64.b64encode(csv.encode()).decode()
+            href = f'<a href="data:file/csv;base64,{b64}" download="output.csv">Download Output CSV</a>'
+            st.markdown(href, unsafe_allow_html=True)
             
 
             # Plot the probabilities