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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +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(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")
 
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[:10], 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")