Update app.py

app.py

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