Spaces:

ryanrwatkins
/

needs

Running

ryanrwatkins commited on Mar 31, 2024

Commit

52de80b

verified ·

1 Parent(s): 4590834

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -475,7 +475,7 @@ chain = ConversationalRetrievalChain.from_llm(
         LLM_provider="Google",api_key=HF_key,temperature=0.1,
         model_name="gemini-pro"),
     memory=create_memory("gemini-pro"),
-    retriever = retriever,
     llm=instantiate_LLM(
         LLM_provider="Google",api_key=HF_key,temperature=0.5,
         model_name="gemini-pro"),

         LLM_provider="Google",api_key=HF_key,temperature=0.1,
         model_name="gemini-pro"),
     memory=create_memory("gemini-pro"),
+    retriever = base_retriever_google,
     llm=instantiate_LLM(
         LLM_provider="Google",api_key=HF_key,temperature=0.5,
         model_name="gemini-pro"),