Update app.py

app.py

@@ -11,6 +11,11 @@ import time
 def load_data(dataset):
     df = pd.read_csv(dataset)
     return df
+def filedownload(df):
+    csv = df.to_csv(index=False)
+    b64 = base64.b64encode(csv.encode()).decode()  # strings <-> bytes conversions
+    href = f'<a href="data:file/csv;base64,{b64}" download="diabete_predictions.csv">Download CSV File</a>'
+    return href
 
 
 st.sidebar.image('images/diabetes.jpg',width=280)
@@ -61,7 +66,7 @@ def main():
                 sns.histplot(x='Glucose',data=data)
                 st.pyplot(fig)
             with tab3:
-                model = pickle.load(open('./models/ada_dump.pkl', 'rb'))
+                model = pickle.load(open('./models/ada_dump.pkl', 'rb'))  # Load the trained model from disk
                 prediction = model.predict(df)
                 pp = pd.DataFrame(prediction, columns=['Prediction'])
                 ndf = pd.concat([df, pp], axis=1)
@@ -70,12 +75,9 @@ def main():
                 st.header("📈🎯 Diabete Risk Prediction")
                 st.subheader("Predictions")
                 st.write(ndf)
-                csv = ndf.to_csv(index=False)
-                b64 = base64.b64encode(csv.encode()).decode()
-
-                href = f'<a href="data:file/csv;base64,{b64}" download="Diabete_Prediction.csv">Download CSV file</a>'
+    
                 if st.button(' 💾 Download csv file'):
-                    st.markdown(href, unsafe_allow_html=True)
+                    st.markdown(filedownload(ndf), unsafe_allow_html=True)