Update README.md

README.md

@@ -105,4 +105,17 @@ Inference can be performed using the provided Streamlit application.
 2. **Select Checkpoints:** Choose the desired checkpoint for each level's model. Checkpoints are saved in the respective `level{n}` directories (e.g., `level1/model` or `level4/level4_step31000`).
 3. **Run Inference:** Click the "Run Inference" button.
 
-The application will output the predicted ID and the corresponding text description for each level of the taxonomy, based on the provided `mapping.csv` file.
+The application will output the predicted ID and the corresponding text description for each level of the taxonomy, based on the provided `mapping.csv` file.
+
+## Visualizations
+
+Here are some images related to the training process:
+
+- ![Level 1 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/blob/main/training/metrics/level-1-train-loss.png)
+- ![Level 1 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/blob/main/training/metrics/level-1-val-loss.png)
+- ![Level 2 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/blob/main/training/metrics/level-2-train-loss.png)
+- ![Level 2 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/blob/main/training/metrics/level-2-val-loss.png)
+- ![Level 3 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/blob/main/training/metrics/level-3-train-loss.png)
+- ![Level 3 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/blob/main/training/metrics/level-3-val-loss.png)
+- ![Level 4 Train Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/blob/main/training/metrics/level-4-train-loss.png)
+- ![Level 4 Validation Loss](https://huggingface.co/dejanseo/ecommerce-taxonomy-classifier/blob/main/training/metrics/level-4-val-loss.png)