allow flax

Just fixing evaluation script, model has not been updated

Update README.md

Update README.md

Update README.md

Update README.md

Update README.md

Update README.md

Update README.md

first version