Kathirsci commited on
Commit
84c1f6a
·
verified ·
1 Parent(s): a446846

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -6,3 +6,10 @@ messages = [
6
  ]
7
  chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
8
  chatbot(messages)
 
 
 
 
 
 
 
 
6
  ]
7
  chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
8
  chatbot(messages)
9
+ from huggingface_hub import snapshot_download
10
+ from pathlib import Path
11
+
12
+ mistral_models_path = Path.home().joinpath('mistral_models', '7B-Instruct-v0.3')
13
+ mistral_models_path.mkdir(parents=True, exist_ok=True)
14
+
15
+ snapshot_download(repo_id="mistralai/Mistral-7B-Instruct-v0.3", allow_patterns=["params.json", "consolidated.safetensors", "tokenizer.model.v3"], local_dir=mistral_models_path)