upload flax model

allow flax

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

initial commit