Upload 2 files

app.py

@@ -9,7 +9,7 @@ import PyPDF2
 import pymupdf
 import tempfile
 import base64
-from tqdm.auto import tqdm
+# from tqdm.auto import tqdm
 import math
 # from transformers import pipeline
 import nltk
@@ -26,6 +26,12 @@ from langchain_community.vectorstores import FAISS
 from langchain.vectorstores import faiss
 from langchain.vectorstores import FAISS
 
+import time
+from time import sleep
+from stqdm import stqdm
+
+
+
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 # if device != 'cuda':
@@ -195,8 +201,10 @@ def get_faiss_semantic_index():
         index_path = "./HuggingFaceEmbeddings.pkl"
         print(index_path)
         # Load embeddings from the pickle file
-        with open(index_path, "rb") as f:
-            faiss_index = pickle.load(f)
+        with stqdm(total=100, desc="Loading pickle file...") as progress_bar:
+            with open(index_path, "rb") as f:
+                faiss_index = pickle.load(f)
+                progress_bar.update(100)
         st.write("Embeddings loaded successfully.")
         return faiss_index
     except Exception as e:

requirements.txt

@@ -15,4 +15,5 @@ nltk
 langchain_community
 langchain
 faiss-gpu
-faiss-cpu
+faiss-cpu
+stqdm