M17idd commited on
Commit
8d6e6bf
·
verified ·
1 Parent(s): 9587c62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -62,6 +62,10 @@ def build_pdf_index():
62
  embeddings = []
63
  batch_size = 16
64
 
 
 
 
 
65
  for i in range(0, len(documents), batch_size):
66
  batch_docs = documents[i:i+batch_size]
67
  batch_texts = [doc.page_content for doc in batch_docs]
@@ -82,6 +86,9 @@ def build_pdf_index():
82
 
83
  embeddings.extend(batch_embeddings)
84
 
 
 
 
85
  # اطمینان که خروجی NumpyArray باشه
86
  embeddings = np.array(embeddings)
87
 
 
62
  embeddings = []
63
  batch_size = 16
64
 
65
+ # اضافه کردن پروگرس بار
66
+ progress_bar = st.progress(0)
67
+ total_batches = len(documents) // batch_size + (1 if len(documents) % batch_size != 0 else 0)
68
+
69
  for i in range(0, len(documents), batch_size):
70
  batch_docs = documents[i:i+batch_size]
71
  batch_texts = [doc.page_content for doc in batch_docs]
 
86
 
87
  embeddings.extend(batch_embeddings)
88
 
89
+ # به‌روزرسانی پروگرس بار
90
+ progress_bar.progress((i + batch_size) / len(documents))
91
+
92
  # اطمینان که خروجی NumpyArray باشه
93
  embeddings = np.array(embeddings)
94