Update README.md

README.md

@@ -21,17 +21,34 @@ TRME (Text Residual Motion Encoder) significantly enhances the process of genera
 - Capable of detailed and complex motion synthesis.
 - Optimized for high diversity and realism in generated motions.
 
-![Model Visualization](https://huggingface.co/rsax/trme/blob/main/motions.png)
+![Model Visualization]
 
-The model has been trained on a novel dataset, CHAD, which includes a comprehensive set of human motion data, enabling it to handle a wide variety of motion generation tasks. Learn more about the training process and model architecture [here](https://huggingface.co/rsax/trme/blob/main/overview.png).
+<figure>
+  <img src="https://huggingface.co/rsax/trme/blob/main/motions.png" alt="TRME">
+  <figcaption>Generated Sequences for MOYO and HumanML3D from T2M-GPT and TRME. The figure provides detailed visualizations of motion sequences generated by our models. The generated motions correspond to two different captions extracted from the HumanML3D and MOYO datasets. Notably, TRME demonstrates superior performance in capturing dependencies across diverse motion classes compared to state-of-the-art models.
+  </figcaption>
+</figure>
+
+The model has been trained on a novel dataset, CHAD, which includes a comprehensive set of human motion data, enabling it to handle a wide variety of motion generation tasks. 
+
+Learn more about the training process and model architecture below:
+
+<figure>
+  <img src="https://huggingface.co/rsax/trme/blob/main/overview.png" alt="overview", width="425", height=480/>
+  <figcaption>Data flow diagram for the TRME model, highlighting the progression from the AMASS database to the creation of the CHAD dataset and subsequent motion generation
+  </figcaption>
+</figure>
 
 ## Example Usage
 
 Here is a demonstration of the model generating a complex human motion from a simple textual description:
 
-"**A person is doing a tree pose.**"
+<figure>
+  <img src="https://huggingface.co/rsax/trme/blob/main/illustration_10.png" alt="illustration", width="425", height=480/>
+  <figcaption>A person is doing a tree pose.
+  </figcaption>
+</figure>
 
-![Generated Motion from Text](https://huggingface.co/rsax/trme/blob/main/illustration_10.png)
 
 ## Datasets