diff --git a/Code/inference.py b/Code/inference.py
index 7689fdd09eae18d5ffa670b8d0fec3bb5971ff3a..4e854d459eda905723bada66624878fda4e2d46b 100644
--- a/Code/inference.py
+++ b/Code/inference.py
@@ -67,7 +67,7 @@ input_as_dataset=preprocess.tokenizer_new(tokenizer, data_sample, max_length=512
 
 # Load model
 
-model=models.WordClassificationModel.from_pretrained("bert-base-uncased")
+model=models.WordClassificationModel("bert-base-uncased")
 
 model_path = "saved_models/bert.pth"
 model = torch.load(model_path, map_location=device)