Update README.md

README.md

@@ -10,14 +10,11 @@ This is a version of the LLama-2-7B-hf model quantized to 4-bit via Half-Quadrat
 
 To run the model, install the HQQ library from https://github.com/mobiusml/hqq and use it as follows:
 ``` Python
-from hqq.models.llama_hf import LlamaHQQ
-import transformers
-
 model_id = 'mobiuslabsgmbh/Llama-2-7b-hf-4bit_g64-HQQ'
-#Load the tokenizer
-tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
-#Load the model
-model = LlamaHQQ.from_quantized(model_id)
+
+from hqq.engine.hf import HQQModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model     = HQQModelForCausalLM.from_quantized(model_id)
 ```
 
 *Limitations*: <br>