Update README.md

README.md

@@ -40,6 +40,17 @@ antspyx
 
 ---
 
+### Model Architecture Overview
+
+![Model Diagram](https://huggingface.co/AI-vBRAIN/pyMEAL/resolve/main/Fig1_TA_NA.png)
+
+*Figure 1. Model architecture for the model having no augmentation and traditional augmentation.*
+
+
+![Model Diagram](https://huggingface.co/AI-vBRAIN/pyMEAL/resolve/main/models/Fig2_BD_CC_FL.png)
+
+*Figure 2. Model architecture for Multi-Stream with a Builder Controller block method (BD), Fusion layer (FL) and Encoder concatenation (CC).*
+
 ## Download Model Files
 
 You can download any `.h5` file directly: