cryogenic22 commited on
Commit
f4bcb1f
·
verified ·
1 Parent(s): f493207

Update utils/database.py

Browse files
Files changed (1) hide show
  1. utils/database.py +1 -1
utils/database.py CHANGED
@@ -340,7 +340,7 @@ def initialize_qa_system(vector_store):
340
  temperature=0.5,
341
  model="gemini-1.5-pro",
342
  api_key=os.environ.get("GEMINI_API_KEY")
343
-
344
  # Create retriever function
345
  retriever = vector_store.as_retriever(search_kwargs={"k": 2})
346
 
 
340
  temperature=0.5,
341
  model="gemini-1.5-pro",
342
  api_key=os.environ.get("GEMINI_API_KEY")
343
+ )
344
  # Create retriever function
345
  retriever = vector_store.as_retriever(search_kwargs={"k": 2})
346