Update README.md

README.md

@@ -5,9 +5,11 @@ license: apache-2.0
 ``` python
 import ctranslate2
 import transformers
+from huggingface_hub import snapshot_download
 
-generator = ctranslate2.Generator("Praise2112/Mistral-7B-v0.1-int8-ct2", device="cuda", compute_type="int8") # GPU
-# generator = ctranslate2.Generator("Praise2112/Mistral-7B-v0.1-int8-ct2", device="cpu", compute_type="int8") #CPU
+model_dir = snapshot_download(repo_id="Praise2112/Mistral-7B-v0.1-int8-ct2")
+generator = ctranslate2.Generator(model_dir, device="cuda", compute_type="int8") # GPU
+# generator = ctranslate2.Generator(model_dir, device="cpu", compute_type="int8") #CPU
 tokenizer = transformers.AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
 
 # Unconditional generation.