Spaces:
Running
Running
Update insight_and_tasks/utils/pandasai_setup.py
Browse files
insight_and_tasks/utils/pandasai_setup.py
CHANGED
@@ -33,7 +33,7 @@ def configure_pandasai(api_key: str, model_name: str = None):
|
|
33 |
|
34 |
try:
|
35 |
llm = LiteLLM(
|
36 |
-
model=
|
37 |
api_key=api_key
|
38 |
)
|
39 |
|
|
|
33 |
|
34 |
try:
|
35 |
llm = LiteLLM(
|
36 |
+
model=DEFAULT_PANDASAI_MODEL , # Use the selected model
|
37 |
api_key=api_key
|
38 |
)
|
39 |
|