Update README.md

README.md

@@ -25,6 +25,17 @@ We have fine-tuned [INDUS Model](https://huggingface.co/nasa-impact/nasa-smd-ibm
 - **Stratified Splitting:** The dataset is split based on `provider-id` to maintain balanced representation across train, validation, and test sets.
 - **Improved Performance:** Focal loss with different focusing parameters (γ) was evaluated, showing significant improvements in weighted precision, recall, F1 score, and Jaccard similarity over cross-entropy loss and previous models.
 
+## Label Mapping During Inference
+
+After obtaining predictions from the model, we can map the predicted label indices to their actual names using the `model.config.id2label` dictionary
+
+```python
+# Example usage
+predicted_indices = [0, 2, 5] # top 3
+predicted_labels = [model.config.id2label[idx] for idx in predicted_indices]
+print(predicted_labels)
+```
+
 
 ## Experiments