HenryY2023 commited on
Commit
4df536e
·
verified ·
1 Parent(s): 75c4275
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -134,9 +134,9 @@ final_answer = FinalAnswerTool()
134
  # num_ctx=8192,
135
  #)
136
  model=LiteLLMModel(
137
- model_id = "gemini-2.0-flash",
138
- api_base = "https://generativelanguage.googleapis.com/v1beta/models",
139
- api_key = api_key,
140
  num_ctx=8192,
141
  )
142
 
 
134
  # num_ctx=8192,
135
  #)
136
  model=LiteLLMModel(
137
+ model_id = "gemini/gemini-2.0-flash-lite",
138
+ #api_base = "https://generativelanguage.googleapis.com/v1beta/models",
139
+ api_key = os.getenv("GEMINI_API_KEY),
140
  num_ctx=8192,
141
  )
142