Update app.py

app.py

@@ -12,7 +12,7 @@ import hashlib
 import pdf2bib
 # Initialize the storage directories if they don't exist
 model_name = "intfloat/e5-large-v2"
-model_kwargs = {'device': 'cuda'}
+model_kwargs = {'device': 'cpu'}
 encode_kwargs = {'normalize_embeddings': False}
 embeddings = HuggingFaceEmbeddings(
     model_name=model_name,
@@ -145,14 +145,14 @@ def edit_metadata_of_selected_pdf(metadata):
 def delete_pdf_and_metadata(metadata):
     # Use a select box for users to choose a PDF to delete
     metadata_dir = st.session_state['metadata_dir']
-    st.write(list(metadata.keys()))
+    #st.write(list(metadata.keys()))
     file_names = list(metadata.keys())
     if not file_names:
         st.write("No PDFs available to delete.")
         return
 
     selected_file_to_delete = st.selectbox('Select a PDF file to delete', file_names)
-    st.write(selected_file_to_delete)
+    #st.write(selected_file_to_delete)
     # Button to delete the PDF and its metadata
     if st.button(f"Delete '{selected_file_to_delete}' and its metadata"):
         # Delete the PDF file