Update app/models.py

app/models.py

@@ -34,10 +34,10 @@ class DataLocation(BaseModel):
                 logger.warning(f"Downloading model from Hugging Face: {self.cloud_uri}")
                 # Download from Hugging Face
                 tokenizer = AutoTokenizer.from_pretrained(
-                    self.cloud_uri, use_auth_token=AUTH_TOKEN
+                    self.cloud_uri, token=AUTH_TOKEN
                 )
                 model = AlbertForQuestionAnswering.from_pretrained(
-                    self.cloud_uri, use_auth_token=AUTH_TOKEN
+                    self.cloud_uri, token=AUTH_TOKEN
                 )
                 # Save the model and tokenizer locally
                 tokenizer.save_pretrained(self.local_path)