Update app.py

app.py

@@ -9,10 +9,10 @@ spider_dataset = load_dataset("spider", split='train')  # Load a subset of the d
 table_names = set()
 column_names = set()
 for item in spider_dataset:
-    for table in item['db']['table_names_original']:
+    for table in item['db_id']:
         table_names.add(table)
-    for column in item['db']['column_names_original']:
-        column_names.add(column[1])
+    for column in item['question']:
+        column_names.add(column)
 
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL")  # Update this to a model fine-tuned on Spider if available