fixed some bugs

models/bloom.py

@@ -67,8 +67,8 @@ def prepare_data(file_path: str):
     documents = []
     for item in parsed:
         document = Document(
-            item["paragraphText"],
-            item["_id"]["$oid"],
+            text=item["paragraphText"],
+            doc_id=item["_id"]["$oid"],
             extra_info={
                 "chapter": item["chapter"],
                 "article": item["article"],

requirements.txt

@@ -8,4 +8,5 @@ openai
 faiss-cpu
 python-dotenv
 streamlit
-huggingface_hub
+huggingface_hub
+xformers