Update README.md

README.md

@@ -3,7 +3,7 @@ tags:
 - generated_from_trainer
 - code
 - coding
-- llama
+- gemma
 model-index:
 - name: gemma-2b-coder
   results: []
@@ -79,21 +79,23 @@ WIP
 
 
 ### Example of usage 👩‍💻
+
+I recommend install the following version of `torch`:
+
+```sh
+pip install "torch>=2.1.1" -U
+```
+
 ```py
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 
-model_id = "mrm8488/llama-2-coder-7b"
+model_id = "MAISAAI/gemma-2b-coder"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 model = AutoModelForCausalLM.from_pretrained(model_id).to("cuda")
 
-def create_prompt(instruction):
-  system = "You are a coding assistant that will help the user to resolve the following instruction:"
-  instruction = "### Instruction: " + instruction
-  return system + "\n" + instruction + "\n\n" + "### Solution:" + "\n"
-
 def generate(
         instruction,
         max_new_tokens=256,