Update app.py

app.py

@@ -13,7 +13,6 @@ def extract_zip(zip_file_path, extract_dir):
     with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
         zip_ref.extractall(extract_dir)
 
-# Function to get embeddings
 # Function to get embeddings
 def get_embeddings(db_dir):
     app = FaceAnalysis(name='buffalo_l')
@@ -75,11 +74,12 @@ def main():
         if st.button("Download Files"):
             st.markdown("### Download names.npy:")
             with open("temp/names.npy", "rb") as file:
-                st.download_button("Download names.npy", file.getvalue(), file_name="names.npy", mime="application/octet-stream")
+                st.download_button("Download names.npy", file.read(), file_name="names.npy", mime="application/octet-stream")
             
             st.markdown("### Download embeddings.npy:")
             with open("temp/embeddings.npy", "rb") as file:
-                st.download_button("Download embeddings.npy", file.getvalue(), file_name="embeddings.npy", mime="application/octet-stream")
+                st.download_button("Download embeddings.npy", file.read(), file_name="embeddings.npy", mime="application/octet-stream")
+
 
         if st.button("Delete Files"):
             delete_files("temp")