Update README.md

README.md

@@ -19,7 +19,7 @@ Developed by [Selector AI](https://www.selector.ai/)
 ```python
 from transformers import BertForSequenceClassification, BertTokenizer
 
-# Load the model and tokenizer from Hugging Face
+# Step 1: Load the model and tokenizer from Hugging Face
 model = BertForSequenceClassification.from_pretrained("rahulm-selector/log-classifier-BERT-v1")
 tokenizer = BertTokenizer.from_pretrained("rahulm-selector/log-classifier-BERT-v1")
 
@@ -41,12 +41,11 @@ with torch.no_grad():
 # Step 4: Get the predicted class (the class with the highest score)
 predicted_class = torch.argmax(logits, dim=1).item()
 
-# Example label mapping (you can load this from a JSON file or config)
+# label mapping (can load from JSON file in repo or config)
 label_mapping = model.config.id2label
 
-# Get the event name from the predicted class
+# Step 5: Get the event name
 predicted_event = label_mapping[predicted_class]
-
 print(f"Predicted Event: {predicted_event}")
 ```