Update inference

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)