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()
|