shahrukhx01
commited on
Commit
•
581f240
1
Parent(s):
dad88c9
Update README.md
Browse files
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)
|