Khachatur Mirijanyan commited on
Commit
13fcfb3
1 Parent(s): 6ea428d

Fix another typo

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,7 +58,7 @@ def get_db(q, tables):
58
  sample_rows_in_table_info=3)
59
  else:
60
  tables.extend(DEFAULT_TABLES)
61
- db = SQLDatabase.from_uri("sqlite:///nba_small.db",
62
  include_tables = tables,
63
  sample_rows_in_table_info=3)
64
  return db
@@ -69,7 +69,7 @@ def answer_question(query):
69
  return('ERROR: Wrong format for getting the big db schema')
70
  if isinstance(query_check, dict):
71
  q = query_check['q']
72
- tables = query_check
73
  if query_check == 'small':
74
  q = query
75
  tables = []
 
58
  sample_rows_in_table_info=3)
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=3)
64
  return db
 
69
  return('ERROR: Wrong format for getting the big db schema')
70
  if isinstance(query_check, dict):
71
  q = query_check['q']
72
+ tables = query_check['tables']
73
  if query_check == 'small':
74
  q = query
75
  tables = []