Update README.md

README.md

@@ -4,20 +4,22 @@ 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
 pipeline_tag: text-generation
 widget:
-- text: '<start_of_turn>user Write a Python function to calculate the nth fibonacci
+- text: >
+    <start_of_turn>user Write a Python function to calculate the nth fibonacci
     number.<end_of_turn> <start_of_turn>model
-
-    '
 inference:
   parameters:
     max_new_tokens: 200
 license_link: https://ai.google.dev/gemma/terms
+base_model:
+- google/codegemma-1.1-7b-it
 ---
 
 # mlx-community/codegemma-1.1-7b-it-4bit
@@ -34,4 +36,4 @@ from mlx_lm import load, generate
 
 model, tokenizer = load("mlx-community/codegemma-1.1-7b-it-4bit")
 response = generate(model, tokenizer, prompt="hello", verbose=True)
-```
+```