Update app.py

app.py

@@ -1,7 +1,7 @@
 
-__import__('pysqlite3')
-import sys
-sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
+#__import__('pysqlite3')
+#import sys
+#sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
 
 
 import os
@@ -39,6 +39,7 @@ from langchain.document_loaders import UnstructuredMarkdownLoader
 from langchain.vectorstores.utils import filter_complex_metadata
 import fitz
 from PIL import Image
+from langchain.vectorstores import FAISS
 
 user_session_id = uuid.uuid4()
 
@@ -46,12 +47,12 @@ logger = get_logger(__name__)
 st.set_page_config(page_title="Document QA by Dono", page_icon="🤖",  )
 st.session_state.disabled = False
 st.title("Document QA by Dono")
-st.markdown(f"""<style>
-            .stApp {{background-image: url("https://media.istockphoto.com/id/450481545/photo/glowing-lightbulb-against-black-background.webp?b=1&s=170667a&w=0&k=20&c=fJ91chWN1UkoKTNUvwgiQwpM80DlRpVC-WlJH_78OvE=");
-                     background-attachment: fixed;
-                     background-size: cover}}
-         </style>
-         """, unsafe_allow_html=True)
+#st.markdown(f"""<style>
+#            .stApp {{background-image: url("https://media.istockphoto.com/id/450481545/photo/glowing-lightbulb-against-black-background.webp?b=1&s=170667a&w=0&k=20&c=fJ91chWN1UkoKTNUvwgiQwpM80DlRpVC-WlJH_78OvE=");
+#                     background-attachment: fixed;
+#                     background-size: cover}}
+#         </style>
+#         """, unsafe_allow_html=True)
 
 DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
 
@@ -74,7 +75,7 @@ def load_model():
     print('embedding done')
     
     #db = Chroma.from_documents(texts, embeddings, persist_directory="/home/user/app/db")
-    db = Chroma.from_documents(texts, embeddings)
+    db = FAISS.from_documents(texts, embeddings)
 
     print('db done')