Spaces:
Paused
Paused
Update utils/database.py
Browse files- 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 |
|