✨🐍 update train script and adds trained models

✨🐍 update train script and adds trained models #17