Upload train.py with huggingface_hub

initial commit