Added model usage and included number of trainable parameters e3ae3d1 verified spolivin commited on Jan 17