Update app.py

app.py

@@ -71,6 +71,13 @@ if uploaded_file is not None:
         # Show cleaned dataset
         st.write("Cleaned Dataset:")
         st.dataframe(df)
+
+        st.download_button(
+    label="Download Cleaned Dataset",
+    data=to_excel(df),  # The cleaned dataset is 'df'
+    file_name="cleaned_dataset.xlsx",
+    mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+)
         
         target = st.selectbox("Select Target Variable", df.columns)
         features = [col for col in df.columns if col != target]