set force_download=True for tokenizer and the model

predict_dept_model.py

@@ -23,8 +23,8 @@ class DepartmentPredictor:
 
         print(" Loading tokenizer and model...")
         # Load tokenizer and model
-        self.tokenizer = AutoTokenizer.from_pretrained(self.model_repo, cache_dir=self.cache_dir)
-        self.model = AutoModelForSequenceClassification.from_pretrained(self.model_repo, cache_dir=self.cache_dir)
+        self.tokenizer = AutoTokenizer.from_pretrained(self.model_repo, cache_dir=self.cache_dir, force_download=True)
+        self.model = AutoModelForSequenceClassification.from_pretrained(self.model_repo, cache_dir=self.cache_dir, force_download=True)
 
         # Create classification pipeline
         self.classifier = pipeline(