pbrenotvinciguerra commited on
Commit
e09cd05
·
1 Parent(s): d7cf6d4

cleaning main

Browse files
Files changed (1) hide show
  1. main.py +3 -3
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')\n",
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, show_progress=True)
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