Update app.py
Browse files
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 |
|