Update app.py

app.py

@@ -13,6 +13,7 @@
 # from chromadb.utils import embedding_functions
 # from langchain.embeddings import SentenceTransformerEmbeddings
 # from langchain.embeddings import HuggingFaceBgeEmbeddings
+from langchain.embeddings import HuggingFaceInstructEmbeddings
 from langchain.document_loaders import (
     CSVLoader,
     DirectoryLoader,
@@ -354,14 +355,19 @@ def main():
                     )
                     st.write("LLM Initialized:")
 
-                    model_name = "BAAI/bge-large-en"
-                    model_kwargs = {'device': 'cpu'}
-                    encode_kwargs = {'normalize_embeddings': False}
-                    embeddings = HuggingFaceBgeEmbeddings(
-                        model_name=model_name,
-                        model_kwargs=model_kwargs,
-                        encode_kwargs=encode_kwargs
-                    )
+                    # model_name = "BAAI/bge-large-en"
+                    # model_kwargs = {'device': 'cpu'}
+                    # encode_kwargs = {'normalize_embeddings': False}
+                    # embeddings = HuggingFaceBgeEmbeddings(
+                    #     model_name=model_name,
+                    #     model_kwargs=model_kwargs,
+                    #     encode_kwargs=encode_kwargs
+                    # )
+
+                    embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-xl",
+                                                      model_kwargs={"device": "cpu"})
+
+
                     
                     # llm = load_model()
                     # prompt = set_custom_prompt()