Update README.md

README.md

@@ -43,7 +43,7 @@ model = transformers.AutoModel.from_pretrained('aehrc/cxrmate-ed', trust_remote_
 model.prepare_data(physionet_dir=physionet_dir, database_dir=database_dir)
 ```
 
-## Generate a report
+## Generate a report:
 
 ```python
 import torch
@@ -123,6 +123,6 @@ Note that you must be a credentialised user to access this dataset.
 Environment requirements can be found here: https://github.com/aehrc/cxrmate-ed/blob/main/requirements.txt.
 
 
-# Training and Code repository:
+# Training:
 
-The code repository, which includes the training pipeline for CXRMate-ED, is available at: https://github.com/aehrc/cxrmate-ed.
+The training pipeline for CXRMate-ED, is available at: https://github.com/aehrc/cxrmate-ed.