change to gensims load function