Update README.md

README.md

@@ -20,14 +20,18 @@ This model is a hierarchical text classifier designed to categorize text into a
 - **Model Developers:** [DEJAN.AI](https://dejan.ai/)
 - **Model Type:** Hierarchical Text Classification
 - **Base Model:** [`albert/albert-base-v2`](https://huggingface.co/albert/albert-base-v2)
-- **Taxonomy Structure:** The model classifies text into a taxonomy with the following structure:
-    - **Level 1:** 21 unique classes
-    - **Level 2:** 193 unique classes
-    - **Level 3:** 1350 unique classes
-    - **Level 4:** 2205 unique classes
-    - **Level 5:** 1387 unique classes
-    - **Level 6:** 399 unique classes
-    - **Level 7:** 50 unique classes
+- **Taxonomy Structure:**
+
+    | Level | Unique Classes |
+    |---|---|
+    | 1 | 21 |
+    | 2 | 193 |
+    | 3 | 1350 |
+    | 4 | 2205 |
+    | 5 | 1387 |
+    | 6 | 399 |
+    | 7 | 50 |
+
 - **Model Architecture:**
     - **Level 1:** Standard sequence classification using `AlbertForSequenceClassification`.
     - **Levels 2-7:** Custom architecture (`TaxonomyClassifier`) where the ALBERT pooled output is concatenated with a one-hot encoded representation of the predicted ID from the previous level before being fed into a linear classification layer.