Commit
·
e09cd05
1
Parent(s):
d7cf6d4
cleaning main
Browse files
main.py
CHANGED
@@ -20,10 +20,10 @@ service_context = ServiceContext.from_defaults(llm=llm, embed_model=embed_model)
|
|
20 |
if "index" in os.list():
|
21 |
storage_context = StorageContext.from_defaults(persist_dir=Path("./index"))
|
22 |
else:
|
23 |
-
UnstructuredReader = download_loader('UnstructuredReader')
|
24 |
-
dir_reader = SimpleDirectoryReader(Path('./docs/'), file_extractor={".pdf": UnstructuredReader()
|
25 |
documents = dir_reader.load_data()
|
26 |
-
index = VectorStoreIndex.from_documents(documents, service_context=service_context
|
27 |
index.storage_context.persist(Path('index'))
|
28 |
storage_context = StorageContext.from_defaults(persist_dir=Path("./index"))
|
29 |
|
|
|
20 |
if "index" in os.list():
|
21 |
storage_context = StorageContext.from_defaults(persist_dir=Path("./index"))
|
22 |
else:
|
23 |
+
UnstructuredReader = download_loader('UnstructuredReader')
|
24 |
+
dir_reader = SimpleDirectoryReader(Path('./docs/'), file_extractor={".pdf": UnstructuredReader()})
|
25 |
documents = dir_reader.load_data()
|
26 |
+
index = VectorStoreIndex.from_documents(documents, service_context=service_context)
|
27 |
index.storage_context.persist(Path('index'))
|
28 |
storage_context = StorageContext.from_defaults(persist_dir=Path("./index"))
|
29 |
|