rishisim commited on
Commit
78585de
1 Parent(s): 7e45574

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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()