pmelnechuk commited on
Commit
b405d4a
·
verified ·
1 Parent(s): 8b9de16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -52,8 +52,7 @@ if __name__=="__main__":
52
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L12-v2")
53
  persist_directory = "./persist_directory"
54
  db = Chroma(persist_directory=persist_directory, embedding_function=embeddings.embed_documents)
55
- vectorstore = Chroma.from_documents(
56
- documentos_fragmentados, embeddings, persist_directory="./chroma_db")
57
  print("Vectorizado terminado")
58
  retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 3})
59
  print("Carga del modelo")
 
52
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L12-v2")
53
  persist_directory = "./persist_directory"
54
  db = Chroma(persist_directory=persist_directory, embedding_function=embeddings.embed_documents)
55
+ vectorstore = Chroma.from_documents(textos, embeddings, persist_directory="./chroma_db")
 
56
  print("Vectorizado terminado")
57
  retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 3})
58
  print("Carga del modelo")