Update README.md

README.md

@@ -46,6 +46,7 @@ This model can generate realistic audios if prompted with the right inference fu
 Please, refer to the [Github Repository](https://github.com/benjaminogbonna/styletts2-finetuned) for complete code.
 
 ```python
+# install libraries
 !sudo apt-get update -y
 !apt-get install build-essential -y
 !pip install torch tensorboard transformers accelerate SoundFile torchaudio librosa phonemizer
@@ -54,6 +55,23 @@ Please, refer to the [Github Repository](https://github.com/benjaminogbonna/styl
 !pip install hf_transfer -qU
 !sudo apt-get install -y espeak-ng
 
+
+# ____________Download model_____________
+model_repo = 'benjaminogbonna/tts_demo_models'
+model_path = 'Models'
+target_files = ['config.yml', 'model_v1.pth']
+local_dir = '.'
+
+import os
+os.makedirs(local_dir, exist_ok=True)
+downloaded_files = []
+for file_name in target_files:
+    file_path = hf_hub_download(repo_id=model_repo, filename=f'{model_path}/{file_name}', local_dir=local_dir)
+    downloaded_files.append(file_path)
+
+print('Downloaded files', downloaded_files)
+
+
 #________________________
 import nltk
 nltk.download('punkt')