Hugging Face's logo

Spaces:
Phoenix21
/
ChatbotLangchain
Sleeping

App Files Community
1
Phoenix21 commited on
Commit
6748fcc
·
verified ·
1 Parent(s): 64d2f90

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -166,7 +166,7 @@ CACHE_SIZE_LIMIT = 1000
166
  # Google Gemini (primary)
167
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
168
  gemini_llm = ChatGoogleGenerativeAI(
169
- model="gemini-1.0-pro",
170
  temperature=0.5,
171
  max_retries=2,
172
  google_api_key=GEMINI_API_KEY
 
166
  # Google Gemini (primary)
167
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
168
  gemini_llm = ChatGoogleGenerativeAI(
169
+ model="gemini-1.5-flash",
170
  temperature=0.5,
171
  max_retries=2,
172
  google_api_key=GEMINI_API_KEY