added smaller size of model to avoid pickle_unpickling error 6b90147 Shruhrid Banthia commited on Apr 2, 2022