fix: update model architecture, features, config to match trained model 136190c verified Arko006 commited on about 5 hours ago