asoria HF staff commited on
Commit
00f2043
1 Parent(s): 249a663

Trying conversational

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from datasets import load_dataset
6
 
7
  def send_prompt(token: str, df: DataFrame, prompt: str) -> str:
8
  llm = Starcoder(api_token=token)
9
- pandas_ai = PandasAI(llm, conversational=False)
10
  return pandas_ai(df, prompt=prompt)
11
 
12
  def get_result(token, dataset, config, split, prompt) -> str:
 
6
 
7
  def send_prompt(token: str, df: DataFrame, prompt: str) -> str:
8
  llm = Starcoder(api_token=token)
9
+ pandas_ai = PandasAI(llm, conversational=True)
10
  return pandas_ai(df, prompt=prompt)
11
 
12
  def get_result(token, dataset, config, split, prompt) -> str: