Fix model name

README.md

@@ -108,11 +108,11 @@ import torch
 from transformers import CanineTokenizer, CanineModel, CanineConfig
 
 # Load tokenizer
-tokenizer = CanineTokenizer.from_pretrained('humbleworth/domain-mlm-epoch-2')
+tokenizer = CanineTokenizer.from_pretrained('humbleworth/domain-mlm')
 
 # Load base CANINE model
-config = CanineConfig.from_pretrained('humbleworth/domain-mlm-epoch-2')
-model = CanineModel.from_pretrained('humbleworth/domain-mlm-epoch-2')
+config = CanineConfig.from_pretrained('humbleworth/domain-mlm')
+model = CanineModel.from_pretrained('humbleworth/domain-mlm')
 
 # Encode a domain
 domain = "example.com"
@@ -138,7 +138,7 @@ from train_mlm import CanineForMaskedLM
 
 # Load model with MLM head
 model = CanineForMaskedLM(config)
-model.canine = CanineModel.from_pretrained('humbleworth/domain-mlm-epoch-2')
+model.canine = CanineModel.from_pretrained('humbleworth/domain-mlm')
 
 # Load MLM head weights
 state_dict = torch.load('training_state.bin', map_location='cpu')
@@ -225,7 +225,7 @@ If you use this model, please cite:
   author={humbleworth},
   year={2025},
   publisher={Hugging Face},
-  howpublished={\url{https://huggingface.co/humbleworth/domain-mlm-epoch-2}}
+  howpublished={\url{https://huggingface.co/humbleworth/domain-mlm}}
 }
 ```