Update README.md

README.md

@@ -46,6 +46,7 @@ import torch
 # Load model and tokenizer
 model = AutoModelForTokenClassification.from_pretrained("oleksiydolgykh/event-message-detector")
 tokenizer = AutoTokenizer.from_pretrained("oleksiydolgykh/event-message-detector")
+tokenizer.truncation_side = "left"
 
 # Example message
 message = "[MESSAGE] [user1]: Let's have a meeting tomorrow at 10 AM."
@@ -59,4 +60,4 @@ with torch.no_grad():
 
 # Process outputs
 logits = outputs.logits
-predictions = torch.argmax(logits, dim=-1)
+predictions = torch.softmax(logits, dim=-1)[:, 1].mean()