Khachatur Mirijanyan
commited on
Commit
•
13fcfb3
1
Parent(s):
6ea428d
Fix another typo
Browse files
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:///
|
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 = []
|