Update app.py

app.py

@@ -10,9 +10,9 @@ db_table_names = set()
 column_names = set()
 for item in spider_dataset:
     db_id = item['db_id']
-    for table in item['schema']['table_names_original']:
+    for table in item['table_names']:
         db_table_names.add((db_id, table))
-    for column in item['schema']['column_names_original']:
+    for column in item['column_names']:
         column_names.add(column[1])
 
 # Load tokenizer and model