Update README.md

README.md

@@ -2,10 +2,10 @@ 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')
-tokenizer = BartTokenizer.from_pretrained('facebook/bart-large-cnn')
+model = BartForConditionalGeneration.from_pretrained('shahrukhx01/distilbart-cnn-12-6-text2sql')
+tokenizer = BartTokenizer.from_pretrained('shahrukhx01/distilbart-cnn-12-6-text2sql')
 
-ARTICLE_TO_SUMMARIZE = "My friends are cool but they eat too many carbs."
+ARTICLE_TO_SUMMARIZE = "what is the temperature of berlin "
 inputs = tokenizer([ARTICLE_TO_SUMMARIZE], max_length=1024, return_tensors='pt')
 
 # Generate Summary