cybersectony
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -44,11 +44,11 @@ pip install torch
|
|
44 |
|
45 |
```python
|
46 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
47 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
48 |
import torch
|
49 |
|
50 |
# Load model and tokenizer
|
51 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
52 |
|
53 |
def predict_email(email_text):
|
54 |
# Preprocess and tokenize
|
|
|
44 |
|
45 |
```python
|
46 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
47 |
+
tokenizer = AutoTokenizer.from_pretrained("cybersectony/phishing-email-detection-distilbert_v2.4.1")
|
48 |
import torch
|
49 |
|
50 |
# Load model and tokenizer
|
51 |
+
model = AutoModelForSequenceClassification.from_pretrained("cybersectony/phishing-email-detection-distilbert_v2.4.1")
|
52 |
|
53 |
def predict_email(email_text):
|
54 |
# Preprocess and tokenize
|