Update README.md

Adds Figures to the model card

README.md

@@ -23,6 +23,11 @@ This is Hummingbird 0.0, a 1B proof-of-concept causal language model based on **
 
 This version of Hummingbird is only meant to demonstrate Efficient Attention for use in causal language modelling. It has been trained on only 15 Billion tokens and is not safeguarded. Therefore, we do not recommend using it as a chatbot.
 
+<div align="center">
+  <img src="figs/Hummingbird.jpg" width="400"/>
+</div>
+
+
 ## Model Details
 
 The model consists of 1.1 Billion parameters with the following specifications:
@@ -36,6 +41,12 @@ The model consists of 1.1 Billion parameters with the following specifications:
 
 The Attention Mechanism used is based on our newly proposed Efficient Attention from our paper, *You Need to Pay Better Attention: Rethinking the Mathematics of Attention Mechanism* ([arXiv:2403.01643](https://arxiv.org/abs/2403.01643)). We have chosen the number of heads to be 1 as an interesting case study since all current LMs use multiple heads.
 
+The loss plot below illustrates the model's performance during training. For comparison, when trained on 15 billion tokens, Hummingbird achieves a slightly lower loss than TinyLlama, a model of similar size.
+<div align="center">
+  <img src="figs/history.png" width="700"/>
+</div>
+
+
 If you use Efficient Attention or Hummingbird, please cite our paper:
 
 ```