gufett0 commited on
Commit
78a0439
1 Parent(s): 2096576

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -13,7 +13,7 @@ import cv2
13
  from llama_index.core import VectorStoreIndex, Settings, SimpleDirectoryReader, ChatPromptTemplate, PromptTemplate, load_index_from_storage, StorageContext
14
  from llama_index.core.node_parser import SentenceSplitter
15
  from llama_index.embeddings.instructor import InstructorEmbedding
16
- from langchain_huggingface import HuggingFaceEmbeddings
17
 
18
 
19
 
@@ -83,8 +83,9 @@ llm_model = None
83
 
84
  Settings.llm = None
85
  #embed_model = InstructorEmbedding(model_name="hkunlp/instructor-base")
86
- embed_model= HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
87
-
 
88
 
89
  documents = SimpleDirectoryReader('./data').load_data()
90
 
 
13
  from llama_index.core import VectorStoreIndex, Settings, SimpleDirectoryReader, ChatPromptTemplate, PromptTemplate, load_index_from_storage, StorageContext
14
  from llama_index.core.node_parser import SentenceSplitter
15
  from llama_index.embeddings.instructor import InstructorEmbedding
16
+ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
17
 
18
 
19
 
 
83
 
84
  Settings.llm = None
85
  #embed_model = InstructorEmbedding(model_name="hkunlp/instructor-base")
86
+ embed_model = HuggingFaceEmbedding(
87
+ model_name="BAAI/bge-small-en-v1.5"
88
+ )
89
 
90
  documents = SimpleDirectoryReader('./data').load_data()
91