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()
|