M17idd commited on
Commit
99ed84f
·
verified ·
1 Parent(s): d287817

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -27,7 +27,7 @@ def load_chunks_and_embeddings():
27
  api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
28
  )
29
 
30
- vectorstore = FAISS.from_documents([], embedding=embeddings) # اول خالی
31
 
32
  # پروگرس بار
33
  progress = st.progress(0, text="🔄 در حال پردازش چانک‌ها...")
@@ -37,7 +37,11 @@ def load_chunks_and_embeddings():
37
  for i in range(0, total, batch_size):
38
  batch_docs = docs[i:i+batch_size]
39
  embeddings_batch = embeddings.embed_documents([doc.page_content for doc in batch_docs])
40
- vectorstore.add_embeddings(embeddings_batch, batch_docs)
 
 
 
 
41
 
42
  progress.progress(min((i+batch_size)/total, 1.0))
43
 
 
27
  api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
28
  )
29
 
30
+ vectorstore = None # هنوز نساختیم
31
 
32
  # پروگرس بار
33
  progress = st.progress(0, text="🔄 در حال پردازش چانک‌ها...")
 
37
  for i in range(0, total, batch_size):
38
  batch_docs = docs[i:i+batch_size]
39
  embeddings_batch = embeddings.embed_documents([doc.page_content for doc in batch_docs])
40
+
41
+ if vectorstore is None:
42
+ vectorstore = FAISS.from_embeddings(embeddings_batch, batch_docs)
43
+ else:
44
+ vectorstore.add_embeddings(embeddings_batch, batch_docs)
45
 
46
  progress.progress(min((i+batch_size)/total, 1.0))
47