Update README.md

README.md

@@ -11,15 +11,15 @@ This is the [Salesforce/codegen2-1B](https://huggingface.co/Salesforce/codegen2-
 
 An example of how to do inference on this model:
 ```python
+from transformers import AutoTokenizer
 from optimum.intel.openvino import OVModelForCausalLM
-from transformers import AutoTokenizer, pipeline
 
-# model_id should be set to either a local directory or a model available on the HuggingFace hub.
-model_id = "helenai/Salesforce-codegen2-1B-ov"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = OVModelForCausalLM.from_pretrained(model_id)
-pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
-result = pipe("My name is Julien and I like to")
-print(result)
+tokenizer = AutoTokenizer.from_pretrained("helenai/Salesforce-codegen2-1B-ov")
+model = OVModelForCausalLM.from_pretrained("helenai/Salesforce-codegen2-1B-ov")
+
+text = "def hello_world():"
+input_ids = tokenizer(text, return_tensors="pt").input_ids
+generated_ids = model.generate(input_ids, max_length=128)
+print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))
 ```