Khachatur Mirijanyan commited on
Commit ·
611b5d4
1
Parent(s): f422003
reduce sample rows
Browse files
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=
|
| 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=
|
| 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()
|