Sunbread's picture
update model & inference
1e53095