Add model card

Put the correct weights in checkpoints

Add the correct weights

Add checkpoints its still not clear

Add trainind data

Add train files

initial commit