Update app.py
Browse files
app.py
CHANGED
@@ -24,12 +24,14 @@ data = loader.load()
|
|
24 |
|
25 |
from langchain_huggingface import HuggingFaceEmbeddings
|
26 |
from langchain_chroma import Chroma
|
27 |
-
from langchain_huggingface.embeddings import HuggingFaceEndpointEmbeddings
|
|
|
28 |
|
29 |
# CHECK MTEB LEADERBOARD & FIND BEST EMBEDDING MODEL
|
30 |
model = "BAAI/bge-m3"
|
31 |
# model = "BAAI/bge-large-en-v1.5"
|
32 |
-
embeddings = HuggingFaceEndpointEmbeddings(model = model)
|
|
|
33 |
|
34 |
vectorstore = Chroma.from_documents(documents = data, embedding = embeddings)
|
35 |
retriever = vectorstore.as_retriever()
|
|
|
24 |
|
25 |
from langchain_huggingface import HuggingFaceEmbeddings
|
26 |
from langchain_chroma import Chroma
|
27 |
+
# from langchain_huggingface.embeddings import HuggingFaceEndpointEmbeddings
|
28 |
+
from langchain_huggingface import HuggingFaceEmbeddings
|
29 |
|
30 |
# CHECK MTEB LEADERBOARD & FIND BEST EMBEDDING MODEL
|
31 |
model = "BAAI/bge-m3"
|
32 |
# model = "BAAI/bge-large-en-v1.5"
|
33 |
+
# embeddings = HuggingFaceEndpointEmbeddings(model = model)
|
34 |
+
embeddings = HuggingFaceEmbeddings(model = model)
|
35 |
|
36 |
vectorstore = Chroma.from_documents(documents = data, embedding = embeddings)
|
37 |
retriever = vectorstore.as_retriever()
|