GuglielmoTor commited on
Commit
2a399a5
·
verified ·
1 Parent(s): 0b3173b

Update insight_and_tasks/utils/pandasai_setup.py

Browse files
insight_and_tasks/utils/pandasai_setup.py CHANGED
@@ -32,7 +32,7 @@ def configure_pandasai(api_key: str, model_name: str = None):
32
 
33
  try:
34
  llm = LiteLLM(
35
- model=selected_model, # Use the selected model
36
  api_key=api_key
37
  )
38
 
 
32
 
33
  try:
34
  llm = LiteLLM(
35
+ model="gemini-2.5-flash-preview-05-20", # Use the selected model
36
  api_key=api_key
37
  )
38