shahrukhx01 commited on
Commit
581f240
1 Parent(s): dad88c9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -13,7 +13,7 @@ tokenizer = BartTokenizer.from_pretrained('shahrukhx01/schema-aware-distilbart-c
13
  ## add NL query with table schema
14
  question = '<col0> Player : text <col1> No. : text <col2> Nationality : text <col3> Position : text <col4> Years in Toronto : text <col5> \ School/Club Team : text </s> What is terrence ross\' nationality'
15
 
16
- inputs = tokenizer([question], max_length=1024, return_tensors='pt')
17
 
18
  # Generate SQL
19
  text_query_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=125, early_stopping=True)
 
13
  ## add NL query with table schema
14
  question = '<col0> Player : text <col1> No. : text <col2> Nationality : text <col3> Position : text <col4> Years in Toronto : text <col5> \ School/Club Team : text </s> What is terrence ross\' nationality'
15
 
16
+ inputs = tokenizer([question], max_length=1024, min_length=0, return_tensors='pt')
17
 
18
  # Generate SQL
19
  text_query_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=125, early_stopping=True)