Update train_model.py

train_model.py

@@ -138,12 +138,16 @@ def main():
     tokenizer.save_pretrained(output_dir)
     
     # Push to Hugging Face Hub
-    model_repo = f"your-username/{args.model_name}"
-    api.create_repo(repo_id=model_repo, private=False, token=hf_token)
+    model_repo = f"your-username/{args.model_name}"  # Replace 'your-username' with your actual username
+    try:
+        api.create_repo(repo_id=model_repo, private=False, token=hf_token)
+    except Exception as e:
+        print(f"Repository might already exist: {e}")
     model.push_to_hub(model_repo, use_auth_token=hf_token)
     tokenizer.push_to_hub(model_repo, use_auth_token=hf_token)
     
     print(f"Model '{args.model_name}' trained and pushed to Hugging Face Hub at '{model_repo}'.")
-
+    
 if __name__ == "__main__":
     main()
+