Update utils.py

utils.py

@@ -44,7 +44,7 @@ def transform_user_question(question, model):
 
 
 def search_document(user_question_embed: list, top_k: int = 1):
-    csv_filename = 'database/document_name_to_embedding.csv'
+    csv_filename = 'document_name_to_embedding.csv'
     if not os.path.exists(csv_filename):
         print("This won't happen!")
         return
@@ -77,7 +77,7 @@ def search_document(user_question_embed: list, top_k: int = 1):
 
 
 def search_document_annoy(user_question_embed: list, top_k: int, metric):
-    csv_filename = 'database/document_name_to_embedding.csv'
+    csv_filename = 'document_name_to_embedding.csv'
     if not os.path.exists(csv_filename):
         print("This won't happen!")
         return
@@ -219,7 +219,7 @@ def main():
         filename_to_embedding_df = get_document_embeddings(path, fn_list)
         final_df = pd.concat([final_df, filename_to_embedding_df], axis=0, ignore_index=True)
 
-    final_df.to_csv('database/document_name_to_embedding.csv')
+    final_df.to_csv('document_name_to_embedding.csv')
 
 
 def parse_downloads_to_title_to_info():