Update README.md

README.md

@@ -4,6 +4,7 @@ Example Usage:
 ```
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
+import torch.nn.functional as F
 
 tokenizer = AutoTokenizer.from_pretrained("will4381/inference_confidence_model")
 model = AutoModelForSequenceClassification.from_pretrained("will4381/inference_confidence_model")
@@ -14,9 +15,11 @@ inference = "The Great Barrier Reef is experiencing a decline in biodiversity du
 inputs = tokenizer(passage, inference, return_tensors="pt", truncation=True, max_length=512, padding="max_length")
 
 model.eval()
+
 with torch.no_grad():
     outputs = model(**inputs)
 
-predicted_confidence = outputs.logits.item()
+predicted_confidence = torch.sigmoid(outputs.logits).item()
+
 print(f"Predicted confidence: {predicted_confidence:.4f}")
 ```