Khachatur Mirijanyan commited on
Commit
611b5d4
1 Parent(s): f422003

reduce sample rows

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,12 +55,12 @@ def check_query(query):
55
  def get_db(q, tables):
56
  if len(tables) == 0:
57
  db = SQLDatabase.from_uri("sqlite:///nba_small.db",
58
- sample_rows_in_table_info=5)
59
  else:
60
  tables.extend(DEFAULT_TABLES)
61
  db = SQLDatabase.from_uri("sqlite:///nba.db",
62
  include_tables = tables,
63
- sample_rows_in_table_info=5)
64
  return db
65
  def answer_question(query):
66
  PROMPT = get_prompt()
 
55
  def get_db(q, tables):
56
  if len(tables) == 0:
57
  db = SQLDatabase.from_uri("sqlite:///nba_small.db",
58
+ sample_rows_in_table_info=2)
59
  else:
60
  tables.extend(DEFAULT_TABLES)
61
  db = SQLDatabase.from_uri("sqlite:///nba.db",
62
  include_tables = tables,
63
+ sample_rows_in_table_info=2)
64
  return db
65
  def answer_question(query):
66
  PROMPT = get_prompt()