Mojo3 commited on
Commit
b9fe0e0
·
1 Parent(s): f92ee23

minor fixes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- Document(page_content=entry["chunk"], metadata={"chunk_index": i})
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