pmelnechuk commited on
Commit
30d0ed3
verified
1 Parent(s): 5637e8e

Prueba para error de tenant

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -49,7 +49,10 @@ if __name__=="__main__":
49
 
50
  # Generaci贸n de embeddings y almacenamiento en base de datos ChromaDB
51
  embeddings = EmbeddingGen("sentence-transformers/all-MiniLM-L12-v2")
52
- vectorstore = Chroma.from_documents(list(tqdm(textos[:10], desc="Procesando documentos", unit="doc")), embeddings)
 
 
 
53
  print("Vectorizado terminado")
54
  retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 3})
55
  print("Carga del modelo")
 
49
 
50
  # Generaci贸n de embeddings y almacenamiento en base de datos ChromaDB
51
  embeddings = EmbeddingGen("sentence-transformers/all-MiniLM-L12-v2")
52
+ persist_directory = "./persist_directory"
53
+ client = chromadb.Client()
54
+ db = Chroma(persist_directory=persist_directory, embedding_function=embeddings.embed_documents, client=client)
55
+ vectorstore = db.from_documents(list(tqdm(textos[:10], desc="Procesando documentos", unit="doc")), embeddings)
56
  print("Vectorizado terminado")
57
  retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 3})
58
  print("Carga del modelo")