Update README.md

README.md

@@ -75,6 +75,23 @@ The model categorizes images into 22 workout classes:
 - **Class 20:** "tricep dips"
 - **Class 21:** "tricep pushdown"
 
+```py
+from datasets import load_dataset
+
+# Load the dataset
+dataset = load_dataset("YOUR-DATASET-HERE")
+
+# Extract unique labels
+labels = dataset["train"].features["label"].names
+
+# Create id2label mapping
+id2label = {str(i): label for i, label in enumerate(labels)}
+
+# Print the mapping
+print(id2label)
+```
+
+
 # **Run with Transformers🤗**
 
 ```python