Update README.md

README.md

@@ -23,16 +23,24 @@ tags:
 ## How to Use
 
 ```python
+import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
-# Load model and tokenizer from Hugging Face Hub
 tokenizer = AutoTokenizer.from_pretrained("harixn/IN-finbert")
 model = AutoModelForSequenceClassification.from_pretrained("harixn/IN-finbert")
 
-# Example inference
 text = "The stock price of XYZ surged today."
 inputs = tokenizer(text, return_tensors="pt")
 outputs = model(**inputs)
+
+# Get probabilities
+probs = torch.softmax(outputs.logits, dim=1)
+print("Probabilities:", probs)
+
+# Get predicted class
+pred_class = torch.argmax(probs, dim=1).item()
+classes = ["negative", "neutral", "positive"]
+print("Predicted class:", classes[pred_class])
 ```
 
 ## Training Data