Spaces:
Sleeping
Sleeping
HusnaManakkot
commited on
Commit
•
5f6cbaa
1
Parent(s):
2922759
Update app.py
Browse files
app.py
CHANGED
@@ -10,10 +10,11 @@ db_table_names = set()
|
|
10 |
column_names = set()
|
11 |
for item in spider_dataset:
|
12 |
db_id = item['db_id']
|
13 |
-
for table in item['
|
14 |
-
|
15 |
-
|
16 |
-
|
|
|
17 |
|
18 |
# Load tokenizer and model
|
19 |
tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL")
|
|
|
10 |
column_names = set()
|
11 |
for item in spider_dataset:
|
12 |
db_id = item['db_id']
|
13 |
+
for table in item['sql']['from']['table_units']:
|
14 |
+
if isinstance(table, list):
|
15 |
+
db_table_names.add((db_id, table[1]))
|
16 |
+
for column in item['sql']['select'][1]:
|
17 |
+
column_names.add(column[1][1])
|
18 |
|
19 |
# Load tokenizer and model
|
20 |
tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL")
|