Update README.md

README.md

@@ -15,8 +15,6 @@ language:
 - **Model Type**: GPTNeoXForCausalLM
 - **Model Training Details**: The model is trained using [crumb/flan-ul2-tinystories](https://huggingface.co/datasets/crumb/flan-ul2-tinystories) which contains around a quarter of a million examples generated from Flan-UL2 (20b) with the prompt "Write a short story using the vocabulary of a first-grader."
 
-![](https://cdn.discordapp.com/attachments/1074346695191711875/1126796435577393213/image.png)
-
 ## Model Description
 
 This model is trained with the specific purpose of generating short narratives using a vocabulary limited to the level of a first-grader. In terms of complexity and language usage, the model is designed to produce simplistic and easily comprehensible text.
@@ -45,4 +43,6 @@ This model serves as a meaningful research tool in exploring the learning tenden
 
 ## Validation and Performance
 
-The model's performance was evaluated using a held-out validation set, which constitutes 1% of the original dataset. During evaluation, the model achieved a loss of 2.284920. During training, the model achieved a loss of 2.647377
+The model's performance was evaluated using a held-out validation set, which constitutes 1% of the original dataset. During evaluation, the model achieved a loss of 2.284920. During training, the model achieved a loss of 2.647377
+
+![](https://cdn.discordapp.com/attachments/1074346695191711875/1126796435577393213/image.png)