Update app.py

app.py

@@ -11,10 +11,10 @@ table_names = set()
 column_names = set()
 for item in spider_dataset:
     db_ids.add(item['db_id'])
-    for table in item['tables']:
-        table_names.add(table['table_name'])
-        for column in table['column_names_original']:
-            column_names.add(column)
+    for table in item['db']['table_names_original']:
+        table_names.add(table)
+    for column in item['db']['column_names_original']:
+        column_names.add(column[1])
 
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL")