add training script and prediction

push to huggingface hug

initial commit