Update README.md

README.md

@@ -39,7 +39,36 @@ This model was trained using a [Phishing Email Detection Dataset](https://huggin
 **Usage Guide**
 **Installation**
 
-```
+```bash
 pip install transformers
 pip install torch
+```
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
+# Load model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained("your-username/model-name")
+model = AutoModelForSequenceClassification.from_pretrained("your-username/model-name")
+
+def predict_phishing(email_text):
+    # Preprocess and tokenize
+    inputs = tokenizer(email_text, return_tensors="pt", truncation=True, max_length=512)
+    
+    # Get prediction
+    with torch.no_grad():
+        outputs = model(**inputs)
+        predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
+    
+    return {
+        "is_phishing": bool(predictions[0][1] > 0.5),
+        "confidence": float(predictions[0][1])
+    }
+
+# Example usage
+email = "Your email text here..."
+result = predict_phishing(email)
+print(f"Is Phishing: {result['is_phishing']}")
+print(f"Confidence: {result['confidence']:.2%}")
 ```