N3tron commited on
Commit
8325815
·
verified ·
1 Parent(s): 1eb160f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -71,14 +71,15 @@ def main():
71
  get_embeddings("temp")
72
  st.success("Embeddings generated successfully!")
73
 
74
- st.markdown("### Download names.npy:")
75
  with open("temp/names.npy", "rb") as file:
76
- st.download_button("Download names.npy", file.read(), file_name="names.npy", mime="application/octet-stream")
77
-
78
- st.markdown("### Download embeddings.npy:")
79
  with open("temp/embeddings.npy", "rb") as file:
80
- st.download_button("Download embeddings.npy", file.read(), file_name="embeddings.npy", mime="application/octet-stream")
81
 
 
82
  delete_files("temp")
83
  st.success("Files deleted successfully!")
84
 
 
71
  get_embeddings("temp")
72
  st.success("Embeddings generated successfully!")
73
 
74
+ if st.button("Download names.npy"):
75
  with open("temp/names.npy", "rb") as file:
76
+ st.download_button("Download names.npy", file.getvalue(), file_name="names.npy", mime="application/octet-stream")
77
+
78
+ if st.button("Download embeddings.npy"):
79
  with open("temp/embeddings.npy", "rb") as file:
80
+ st.download_button("Download embeddings.npy", file.getvalue(), file_name="embeddings.npy", mime="application/octet-stream")
81
 
82
+ if st.button("Delete Files"):
83
  delete_files("temp")
84
  st.success("Files deleted successfully!")
85