Update README.md

README.md

@@ -1,3 +1,5 @@
+The distilbart-cnn-12-6-text2sql is fine-tuned on WIKISQL dataset.
+```python
 from transformers import BartTokenizer, BartForConditionalGeneration, BartConfig
 
 model = BartForConditionalGeneration.from_pretrained('facebook/bart-large-cnn')
@@ -8,4 +10,5 @@ inputs = tokenizer([ARTICLE_TO_SUMMARIZE], max_length=1024, return_tensors='pt')
 
 # Generate Summary
 summary_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=5, early_stopping=True)
-print([tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in summary_ids])
+print([tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in summary_ids])
+```