cybersectony commited on
Commit
62d0f99
·
verified ·
1 Parent(s): 3a0bb7a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -44,11 +44,11 @@ pip install torch
44
 
45
  ```python
46
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
47
- tokenizer = AutoTokenizer.from_pretrained("your-username/model-name")
48
  import torch
49
 
50
  # Load model and tokenizer
51
- model = AutoModelForSequenceClassification.from_pretrained("your-username/model-name")
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