Update README.md

README.md

@@ -61,10 +61,10 @@ The following hyperparameters were used during training:
 
 ### How to Get Started with the Model
 Use the code below to get started with the model. You can adjust hyperparameters to fit on your data.
-'''
+```python
 def diary_summary(text):
   input_ids = tokenizer.encode(text, return_tensors = 'pt').to(device)
   summary_text_ids = model.generate(input_ids = input_ids, bos_token_id = model.config.bos_token_id, eos_token_id = model.config.eos_token_id,
                                     length_penalty = 2.0, max_length = 150, num_beams = 2)
   return tokenizer.decode(summary_text_ids[0], skip_special_tokens = True)
-'''
+```