shahrukhx01
commited on
Commit
•
26632ab
1
Parent(s):
a5391fc
Update README.md
Browse files
README.md
CHANGED
@@ -1,4 +1,15 @@
|
|
1 |
```python
|
|
|
2 |
model = BartForConditionalGeneration.from_pretrained('shahrukhx01/schema-aware-distilbart-cnn-12-6-text2sql')
|
3 |
tokenizer = BartTokenizer.from_pretrained('shahrukhx01/schema-aware-distilbart-cnn-12-6-text2sql')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
```
|
|
|
1 |
```python
|
2 |
+
from transformers import BartTokenizer, BartForConditionalGeneration, BartConfig
|
3 |
model = BartForConditionalGeneration.from_pretrained('shahrukhx01/schema-aware-distilbart-cnn-12-6-text2sql')
|
4 |
tokenizer = BartTokenizer.from_pretrained('shahrukhx01/schema-aware-distilbart-cnn-12-6-text2sql')
|
5 |
+
## add NL query with table schema
|
6 |
+
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'
|
7 |
+
|
8 |
+
'
|
9 |
+
inputs = tokenizer([question], max_length=1024, return_tensors='pt')
|
10 |
+
|
11 |
+
# Generate SQL
|
12 |
+
text_query_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=125, early_stopping=True)
|
13 |
+
prediction = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in text_query_ids][0]
|
14 |
+
print(prediction)
|
15 |
```
|