Spaces:
Running
Running
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=
|
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 |
|