Spaces:
Sleeping
Sleeping
Prueba para error de tenant
Browse files
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 |
-
|
|
|
|
|
|
|
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")
|