sanjeevbora commited on
Commit
125b60f
·
verified ·
1 Parent(s): 5cf3079
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
16
  model_kwargs = {"device": device}
17
  embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
18
 
19
- loader = DirectoryLoader('./companies', glob="**/*.pdf", recursive=True, use_multithreading=True)
20
  docs = loader.load()
21
  vectordb = Chroma.from_documents(documents=docs, embedding=embeddings, persist_directory="companies_db")
22
  books_db = Chroma(persist_directory="./companies_db", embedding_function=embeddings)
 
16
  model_kwargs = {"device": device}
17
  embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
18
 
19
+ loader = DirectoryLoader('./example', glob="**/*.pdf", recursive=True, use_multithreading=True)
20
  docs = loader.load()
21
  vectordb = Chroma.from_documents(documents=docs, embedding=embeddings, persist_directory="companies_db")
22
  books_db = Chroma(persist_directory="./companies_db", embedding_function=embeddings)