Spaces:
Sleeping
Sleeping
minor fixes
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import torch
|
|
7 |
import time
|
8 |
from sentence_transformers import SentenceTransformer
|
9 |
from langchain.vectorstores import Chroma
|
10 |
-
from langchain.docstore.document import Document
|
11 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
12 |
|
13 |
|
@@ -104,7 +104,7 @@ embeddings = embed_chunks(chunks)
|
|
104 |
# Step 3: Prepare documents (unchanged)
|
105 |
def prepare_documents_for_chroma(embeddings):
|
106 |
return [
|
107 |
-
|
108 |
for i, entry in enumerate(embeddings, start=1)
|
109 |
]
|
110 |
|
|
|
7 |
import time
|
8 |
from sentence_transformers import SentenceTransformer
|
9 |
from langchain.vectorstores import Chroma
|
10 |
+
from langchain.docstore.document import Document as Document2
|
11 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
12 |
|
13 |
|
|
|
104 |
# Step 3: Prepare documents (unchanged)
|
105 |
def prepare_documents_for_chroma(embeddings):
|
106 |
return [
|
107 |
+
Document2(page_content=entry["chunk"], metadata={"chunk_index": i})
|
108 |
for i, entry in enumerate(embeddings, start=1)
|
109 |
]
|
110 |
|