Update README.md

README.md

@@ -41,18 +41,13 @@ Via pip: `pip install llm-rs`
 
 #### Run inference
 ```python
-from llm_rs import Mpt
-from huggingface_hub import hf_hub_download
+from llm_rs import AutoModel
 
-#Download the model
-hf_hub_download(repo_id="LLukas22/mpt-7b-ggml", filename="mpt-7b-q4_0-ggjt.bin", local_dir=".")
-
-#Load the model
-model = Mpt("mpt-7b-q4_0-ggjt.bin")
+#Load the model, define any model you like from the list above as the `model_file`
+model = AutoModel.from_pretrained("Rustformers/mpt-7b-ggml",model_file="mpt-7b-q4_0-ggjt.bin")
 
 #Generate
 print(model.generate("The meaning of life is"))
-```
 
 ### Rust via [Rustformers/llm](https://github.com/rustformers/llm):