Update app.py
Browse files
app.py
CHANGED
@@ -110,14 +110,11 @@ def initialize_database(list_file_obj):
|
|
110 |
collection_name[-1] = 'Z'
|
111 |
# print('list_file_path: ', list_file_path)
|
112 |
print('Collection name: ', collection_name)
|
113 |
-
progress(0.25, desc="Loading document...")
|
114 |
# Load document and create splits
|
115 |
doc_splits = load_doc(list_file_path, chunk_size, chunk_overlap)
|
116 |
# Create or load vector database
|
117 |
-
progress(0.5, desc="Generating vector database...")
|
118 |
# global vector_db
|
119 |
vector_db = create_db(doc_splits, collection_name)
|
120 |
-
progress(0.9, desc="Done!")
|
121 |
return vector_db, collection_name
|
122 |
|
123 |
|
|
|
110 |
collection_name[-1] = 'Z'
|
111 |
# print('list_file_path: ', list_file_path)
|
112 |
print('Collection name: ', collection_name)
|
|
|
113 |
# Load document and create splits
|
114 |
doc_splits = load_doc(list_file_path, chunk_size, chunk_overlap)
|
115 |
# Create or load vector database
|
|
|
116 |
# global vector_db
|
117 |
vector_db = create_db(doc_splits, collection_name)
|
|
|
118 |
return vector_db, collection_name
|
119 |
|
120 |
|