Update train.py

train.py

@@ -98,7 +98,8 @@ repo_url = api.create_repo(repo_id=repo_name, token=api_token, exist_ok=True)
 # Initialize the repository
 repo = Repository(local_dir="hf_model_repo", clone_from=repo_url, use_auth_token=api_token)
 
-# Directly push the model file to the Hugging Face Hub from the dataset directory
-repo.push_to_hub(commit_message="Upload trained model", repo_id=repo_name)
-
+# Move the model file to the repository directory
+os.rename(saved_model_path, os.path.join("hf_model_repo", "model.joblib"))
 
+# Push the model to the Hugging Face Hub
+repo.push_to_hub(commit_message="Upload trained model")