Update README.md

README.md

@@ -10,4 +10,37 @@ pinned: false
 license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# RATCHET: RAdiological Text Captioning for Human Examined Thoraxes
+
+RATCHET is a Medical Transformer for Chest X-ray Diagnosis and Reporting. Based on the architecture featured in [Attention Is All You Need](https://papers.nips.cc/paper/7181-attention-is-all-you-need.pdf). This network is trained and validated on the MIMIC-CXR v2.0.0 dataset.
+
+
+### Architecture
+
+<p align="center">
+  <img src="assets/model_transformer.png" alt="RATCHET Architecture" width="300"/>
+</p>
+
+
+### Results
+
+<p align="center">
+  &nbsp;&nbsp;&nbsp;&nbsp;
+  <img src="https://upload.wikimedia.org/wikipedia/commons/7/7a/Cardiomegally.PNG" alt="Cardiomegaly" height="300"/>
+  &nbsp;&nbsp;&nbsp;&nbsp;
+  &nbsp;&nbsp;&nbsp;&nbsp;
+  <img src="assets/attn_plot.png" alt="Cardiomegaly Attention Plot" height="300"/>
+  &nbsp;&nbsp;&nbsp;&nbsp;
+</p>
+
+
+##### Generated Text: 
+
+> In comparison with the study of \_\_\_, there is little overall change. Again there is substantial enlargement of the cardiac silhouette with a dual-channel pacer device in place. No evidence of vascular congestion or acute focal pneumonia. Blunting of the costophrenic angles is again seen.
+
+
+### More Examples
+
+<p align="center">
+  <img src="assets/examples.png" alt="More Captioning Examples" max-height="400"/>
+</p>