Add more sample rows

app.py

@@ -27,7 +27,7 @@ def get_prompt():
     Only use the following tables:
 
     {table_info}
-    
+
     Question: {input}"""
 
     PROMPT = PromptTemplate(
@@ -55,12 +55,12 @@ def check_query(query):
 def get_db(q, tables):
     if len(tables) == 0:
         db = SQLDatabase.from_uri("sqlite:///nba_small.db",
-                                  sample_rows_in_table_info=3)
+                                  sample_rows_in_table_info=5)
     else:
         tables.extend(DEFAULT_TABLES)
         db = SQLDatabase.from_uri("sqlite:///nba.db",
                                   include_tables = tables,
-                                  sample_rows_in_table_info=3)
+                                  sample_rows_in_table_info=5)
     return db
 def answer_question(query):
     PROMPT = get_prompt()