Update app.py
Browse files
app.py
CHANGED
|
@@ -71,14 +71,15 @@ def main():
|
|
| 71 |
get_embeddings("temp")
|
| 72 |
st.success("Embeddings generated successfully!")
|
| 73 |
|
| 74 |
-
|
| 75 |
with open("temp/names.npy", "rb") as file:
|
| 76 |
-
st.download_button("Download names.npy", file.
|
| 77 |
-
|
| 78 |
-
|
| 79 |
with open("temp/embeddings.npy", "rb") as file:
|
| 80 |
-
st.download_button("Download embeddings.npy", file.
|
| 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 |
|