Update README.md

README.md

@@ -8,7 +8,7 @@ tags:
 datasets:
 - wikisql
 widget:
-- text: "translate English to Sql: What was the number of race that Kevin Curtain won? </s> c0 | number <eom>  v4 | Kevin Curtain </s> c0 | No <eom> c1 | Date <eom> c2 | Round <eom> c3 | Circuit <eom> c4 | Pole_Position <eom> c5 | Fastest_Lap <eom> c6 | Race_winner <eom> c7 | Report"
+- text: "translate English to Sql: What Branding does WRKO calls use? </s> c2 | Branding <eom> c0 | calls <eom> v0 | WRKO </s> c0 | Calls <eom> c1 | Frequency <eom> c2 | Branding <eom> c3 | Format <eom> c4 | Market/Rank <eom> c5 | Timeslot <eom> c6 | Group_owner"
 ---
 
 ## Paper
@@ -43,9 +43,9 @@ model = AutoModelForSeq2SeqLM.from_pretrained("course5i/NatSight-t5-small-wikisq
 
 # define input
 prefix = "translate English to Sql: "
-raw_nat_query = "What was the number of race that Kevin Curtain won?"
-query_mention_schema = "c0 | number <eom>  v4 | Kevin Curtain"
-table_header_schema = "c0 | No <eom> c1 | Date <eom> c2 | Round <eom> c3 | Circuit <eom> c4 | Pole_Position <eom> c5 | Fastest_Lap <eom> c6 | Race_winner <eom> c7 | Report"
+raw_nat_query = "What Branding does WRKO calls use?"
+query_mention_schema = "c2 | Branding <eom> c0 | calls <eom> v0 | WRKO"
+table_header_schema = "c0 | Calls <eom> c1 | Frequency <eom> c2 | Branding <eom> c3 | Format <eom> c4 | Market/Rank <eom> c5 | Timeslot <eom> c6 | Group_owner"
 
 encoder_input = prefix + raw_nat_query + " </s> " + query_mention_schema + " </s> " + table_header_schema
 input_ids = tokenizer.encode(encoder_input, return_tensors="pt", add_special_tokens=True)