upload flax model

allow flax

First version of bert-base-uncased model finetuned on SST

initial commit