fix

app.py

@@ -134,9 +134,7 @@ if st.button('Generate Dataset'):  # Generate the samples
 
     csv = dataframe.to_csv('2D_Lattice.csv')
 
-    file = '2D_Lattice.csv'
-
-    st.download_button("Download Dataset", file, file_name='2D_Lattice.csv')
+    st.download_button("Download Dataset", csv, file_name='2D_Lattice.csv')
 '''
 
 # food = load_dataset("cmudrc/2d-lattices", split="train[:15]")  # Loads the training data samples