chemouda commited on
Commit
fecdcde
·
verified ·
1 Parent(s): 5053785

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -43,14 +43,14 @@ from langchain.retrievers import ContextualCompressionRetriever
43
 
44
  load_dotenv()
45
 
46
- os.getenv("COHERE_API_KEY")
47
  #os.getenv("GROQ_API_KEY")
48
- #os.getenv("OPENAI_API_KEY")
49
 
50
  embeddings = OpenAIEmbeddings(model="text-embedding-3-large")
51
  #embeddings_cohere = CohereEmbeddings(model="embed-multilingual-v3.0")
52
 
53
- def load_vectorstore(index_name, embeddings, k=100):
54
  return FAISS.load_local(index_name, embeddings, allow_dangerous_deserialization=True).as_retriever(
55
  search_kwargs={ "k": k}
56
  )
 
43
 
44
  load_dotenv()
45
 
46
+ #os.getenv("COHERE_API_KEY")
47
  #os.getenv("GROQ_API_KEY")
48
+ os.getenv("OPENAI_API_KEY")
49
 
50
  embeddings = OpenAIEmbeddings(model="text-embedding-3-large")
51
  #embeddings_cohere = CohereEmbeddings(model="embed-multilingual-v3.0")
52
 
53
+ def load_vectorstore(index_name, embeddings, k=20):
54
  return FAISS.load_local(index_name, embeddings, allow_dangerous_deserialization=True).as_retriever(
55
  search_kwargs={ "k": k}
56
  )