Spaces:
Paused
Paused
example
Browse files
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('./
|
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)
|