Update app.py

app.py

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