arithescientist commited on
Commit
9bff135
·
verified ·
1 Parent(s): 1ace948

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -80,7 +80,8 @@ def parse_user_question(question):
80
 
81
  Answer:
82
  """
83
- response = llm(parsing_prompt)
 
84
  try:
85
  parsed_query = json.loads(response)
86
  return parsed_query
@@ -164,7 +165,8 @@ def generate_data_summary():
164
 
165
  Provide a concise summary of the dataset, highlighting key statistics and any notable observations.
166
  """
167
- summary = llm(summary_prompt)
 
168
  return summary
169
 
170
  # Step 5: Define the callback function
 
80
 
81
  Answer:
82
  """
83
+ # Use llm.predict instead of llm()
84
+ response = llm.predict(parsing_prompt)
85
  try:
86
  parsed_query = json.loads(response)
87
  return parsed_query
 
165
 
166
  Provide a concise summary of the dataset, highlighting key statistics and any notable observations.
167
  """
168
+ # Use llm.predict instead of llm()
169
+ summary = llm.predict(summary_prompt)
170
  return summary
171
 
172
  # Step 5: Define the callback function