Update README.md

README.md

@@ -226,7 +226,7 @@ This is a simple example of how to use **Granite-34B-Code-Instruct** model.
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # or "cpu"
-model_path = "granite-8b-code-instruct"
+model_path = "ibm-granite/granite-34b-code-instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 # drop device_map if running on CPU
 model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)
@@ -250,6 +250,7 @@ for i in output:
     print(i)
 ```
 
+
 <!-- TO DO: Check this part -->
 ## Training Data
 Granite Code Instruct models are trained on the following types of data.