bug fixing

Author: haoheliu

README.md

@@ -14,20 +14,28 @@ This package provides:
 
 Please visit [demo page](https://haoheliu.github.io/demopage-voicefixer/) to view what voicefixer can do.
 
+## Usage
 
-You can test audio samples on your desktop by running website (powered by [streamlit](https://streamlit.io/))
-
-```
-# install additional web package
-$ pip install streamlit
-# you can run after installing voice_fixer package 
-$ streamlit run test/streamlit.py
+Install voicefixer first:
+```shell script
+pip install voicefixer
 ```
 
+### Desktop App
 
-## Usage
+You can test audio samples on your desktop by running website (powered by [streamlit](https://streamlit.io/))
+```shell script
+# Install additional web package
+pip install streamlit
+# Run streamlit 
+streamlit run test/streamlit.py
+```
+**Important:** When you run the above command for the first time, the web page may leave blank for several minutes for downloading models. You can checkout the terminal for downloading progresses.  
+ 
 
-### Basic example:
+### Python interface
+
+Basic example: 
 
 ```python
 # Will automatically download model parameters.
@@ -54,8 +62,8 @@ voicefixer.restore(input="", # input wav file path
                    cuda=False, # whether to use gpu acceleration
                    mode = 2) # You can try out mode 0, 1, 2 to find out the best result
 
-
-
+# Another similar function
+# voicefixer.restore_inmem()
 
 # Universal speaker independent vocoder
 vocoder = Vocoder(sample_rate=44100) # Only 44100 sampling rate is supported.

test/streamlit.py

@@ -37,6 +37,8 @@ def init_voicefixer():
         if is_cuda != list(voice_fixer._model.parameters())[0].is_cuda:
             device = 'cuda' if is_cuda else 'cpu'
             voice_fixer._model = voice_fixer._model.to(device)
+    else:
+        is_cuda=False
 
     t1 = time.time()
 

test/test.py

@@ -14,6 +14,16 @@
 
 voicefixer = VoiceFixer()
 
-voicefixer.restore(input="/Users/liuhaohe/Downloads/lieshi_short.wav",
-                   output="/Users/liuhaohe/Downloads/lieshi_short.wav",
+voicefixer.restore(input="/Users/liuhaohe/Downloads/vocals.wav",
+                   output="/Users/liuhaohe/Downloads/vocals_mode_0.wav",
+                   cuda=False,mode=0)
+
+
+voicefixer.restore(input="/Users/liuhaohe/Downloads/vocals.wav",
+                   output="/Users/liuhaohe/Downloads/vocals_mode_1.wav",
+                   cuda=False,mode=1)
+
+
+voicefixer.restore(input="/Users/liuhaohe/Downloads/vocals.wav",
+                   output="/Users/liuhaohe/Downloads/vocals_mode_2.wav",
                    cuda=False,mode=2)