Updated README

bndzor · Aug 29, 2023 · 9bbc57d · 9bbc57d
1 parent c4c247b
commit 9bbc57d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/README-ZH.md b/README-ZH.md
@@ -221,6 +221,12 @@ make_prompt(name="paimon", audio_prompt_path="paimon_prompt.wav")
 ```
 来尝试一下刚刚做好的音频提示吧！
 ```python
+from utils.generation import SAMPLE_RATE, generate_audio, preload_models
+from scipy.io.wavfile import write as write_wav
+
+# download and load all models
+preload_models()
+
 text_prompt = """
 Hey, Traveler, Listen to this, This machine has taken my voice, and now it can talk just like me!
 """

diff --git a/README.md b/README.md
@@ -221,6 +221,12 @@ make_prompt(name="paimon", audio_prompt_path="paimon_prompt.wav")
 ```
 Now let's try out the prompt we've just made!
 ```python
+from utils.generation import SAMPLE_RATE, generate_audio, preload_models
+from scipy.io.wavfile import write as write_wav
+
+# download and load all models
+preload_models()
+
 text_prompt = """
 Hey, Traveler, Listen to this, This machine has taken my voice, and now it can talk just like me!
 """