Spaces:
Running
on
T4
Running
on
T4
Update auditqa/doc_process.py
Browse files- auditqa/doc_process.py +2 -3
auditqa/doc_process.py
CHANGED
|
@@ -86,15 +86,14 @@ def process_pdf():
|
|
| 86 |
qdrant_collections[file] = Qdrant.from_documents(
|
| 87 |
value,
|
| 88 |
embeddings,
|
| 89 |
-
path=f"./data/
|
| 90 |
collection_name=file,
|
| 91 |
)
|
| 92 |
|
| 93 |
print("done")
|
| 94 |
return qdrant_collections
|
| 95 |
|
| 96 |
-
def get_local_qdrant(name):
|
| 97 |
-
client = QdrantClient(path=f"./data/{name}")
|
| 98 |
print(client.get_collections())
|
| 99 |
embeddings = HuggingFaceEmbeddings(
|
| 100 |
model_kwargs = {'device': device},
|
|
|
|
| 86 |
qdrant_collections[file] = Qdrant.from_documents(
|
| 87 |
value,
|
| 88 |
embeddings,
|
| 89 |
+
path=f"./data/local_qdrant",
|
| 90 |
collection_name=file,
|
| 91 |
)
|
| 92 |
|
| 93 |
print("done")
|
| 94 |
return qdrant_collections
|
| 95 |
|
| 96 |
+
def get_local_qdrant(client, name):
|
|
|
|
| 97 |
print(client.get_collections())
|
| 98 |
embeddings = HuggingFaceEmbeddings(
|
| 99 |
model_kwargs = {'device': device},
|