Update README.md

README.md

@@ -38,6 +38,7 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 
 Use the code below to get started with the model.
 
+```
 from transformers import BartForConditionalGeneration, AutoTokenizer
 
 model = BartForConditionalGeneration.from_pretrained('yousefg/Academ-0.5')
@@ -65,6 +66,7 @@ attention_mask = torch.tensor(batch['attention_mask']).unsqueeze(0).to(device)
 
 summary = get_summary(input_ids, attention_mask, 1654)
 print(summary)
+```
 
 ## Training Details