fix script

run.sh

@@ -1,6 +1,7 @@
 #!/usr/bin/env bash
 python3 run_mlm_flax.py \
-    --output_dir="flax_model.msgpack" \
+    --model_name_or_path="flax_model.msgpack" \
+    --output_dir="./" \
     --model_type="roberta" \
     --config_name="./" \
     --tokenizer_name="./" \