Update README.md

README.md

@@ -109,13 +109,34 @@ The application will output the predicted ID and the corresponding text descript
 
 ## Visualizations
 
-Here are some images related to the training process:
-
-- ![Level 1 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-1-train-loss.png)
-- ![Level 1 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-1-val-loss.png)
-- ![Level 2 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-2-train-loss.png)
-- ![Level 2 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-2-val-loss.png)
-- ![Level 3 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-3-train-loss.png)
-- ![Level 3 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-3-val-loss.png)
-- ![Level 4 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-4-train-loss.png)
-- ![Level 4 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-4-val-loss.png)
+### Level 1: Training Loss
+![Level 1 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-1-train-loss.png)
+This graph shows the training loss over the steps for Level 1, demonstrating a significant drop in loss during the initial training period.
+
+### Level 1: Validation Loss
+![Level 1 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-1-val-loss.png)
+This graph illustrates the validation loss progression over training steps for Level 1, showing steady improvement.
+
+### Level 2: Training Loss
+![Level 2 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-2-train-loss.png)
+Here we see the training loss for Level 2, which also shows a significant decrease early on in training.
+
+### Level 2: Validation Loss
+![Level 2 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-2-val-loss.png)
+The validation loss for Level 2 shows consistent reduction as training progresses.
+
+### Level 3: Training Loss
+![Level 3 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-3-train-loss.png)
+This graph displays the training loss for Level 3, where training stabilizes after an initial drop.
+
+### Level 3: Validation Loss
+![Level 3 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-3-val-loss.png)
+The validation loss for Level 3, demonstrating steady improvements as the model converges.
+
+### Level 4: Training Loss
+![Level 4 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-4-train-loss.png)
+The training loss for Level 4 is plotted here, showing the effects of high-dimensional input features at this level.
+
+### Level 4: Validation Loss
+![Level 4 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/resolve/main/training/metrics/level-4-val-loss.png)
+Finally, the validation loss for Level 4 is shown, where training seems to stabilize after a longer period.