Update README.md

README.md

@@ -8,6 +8,7 @@ tags:
 license: apache-2.0
 datasets:
 - bvk/SMS-spam
+- corbt/enron-emails
 language:
 - en
 metrics:
@@ -66,7 +67,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipe
 
 # Load the model and tokenizer
 model_name = "kenbaker-gif/Email-Spam-Classifier"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(Email_Spam_Classifier)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
 # Create a text-classification pipeline
@@ -159,7 +160,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipe
 # Load model and tokenizer
 model_name = "kenbaker-gif/Email-Spam-Classifier"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(Email_Spam_Classifier)
 
 # Create pipeline
 classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
@@ -233,4 +234,4 @@ Ainebyona Abuabker
 ## Model Card Contact
 - Name: Ainebyona Abubaker
 - Email: ainebyonabubaker@proton.me
-- GitHub: https://github.com/kenbaker-gif
+- GitHub: https://github.com/kenbaker-gif