Update train_model.py

train_model.py

@@ -137,7 +137,7 @@ def main():
 
     # Initialize Hugging Face API
     api = HfApi()
-    hf_token = HfFolder.get_token()
+    hf_token = os.getenv('HF_API_TOKEN')
     if not hf_token:
         logging.error("HF_API_TOKEN is not set. Please set it as an environment variable.")
         raise ValueError("HF_API_TOKEN is not set.")