Update code snippet

README.md

@@ -18,7 +18,6 @@ Weights have been converted to `float16` from the original `bfloat16` type, beca
 How to use with [MLX](https://github.com/ml-explore/mlx).
 
 ```bash
-
 # Install mlx, mlx-examples, huggingface-cli
 pip install mlx
 pip install huggingface_hub hf_transfer
@@ -29,7 +28,7 @@ export HF_HUB_ENABLE_HF_TRANSFER=1
 huggingface-cli download --local-dir CodeLlama-7b-Python-mlx mlx-llama/CodeLlama-7b-Python-mlx
 
 # Run example
-python mlx-examples/llama/llama.py CodeLlama-7b-Python-mlx/ CodeLlama-7b-Python-mlx/tokenizer.model "def fibonacci("
+python mlx-examples/llama/llama.py --prompt "def fibonacci(n):" CodeLlama-7b-Python-mlx/ CodeLlama-7b-Python-mlx/tokenizer.model --max-tokens 200
 ```
 
 Please, refer to the [original model card](https://github.com/facebookresearch/codellama/blob/main/MODEL_CARD.md) for details on CodeLlama.