Update README.md

README.md

@@ -4,11 +4,14 @@ library_name: transformers
 tags:
 - mlx
 extra_gated_heading: Access CodeGemma on Hugging Face
-extra_gated_prompt: To access CodeGemma on Hugging Face, you’re required to review
-  and agree to Google’s usage license. To do this, please ensure you’re logged-in
-  to Hugging Face and click below. Requests are processed immediately.
+extra_gated_prompt: >-
+  To access CodeGemma on Hugging Face, you’re required to review and agree to
+  Google’s usage license. To do this, please ensure you’re logged-in to Hugging
+  Face and click below. Requests are processed immediately.
 extra_gated_button_content: Acknowledge license
 license_link: https://ai.google.dev/gemma/terms
+base_model:
+- google/codegemma-1.1-2b
 ---
 
 # mlx-community/codegemma-1.1-2b-8bit
@@ -25,4 +28,4 @@ from mlx_lm import load, generate
 
 model, tokenizer = load("mlx-community/codegemma-1.1-2b-8bit")
 response = generate(model, tokenizer, prompt="hello", verbose=True)
-```
+```