Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
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=
|
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 |
|