Spaces:
Runtime error
Runtime error
StormblessedKal
commited on
Commit
•
96979b3
1
Parent(s):
49535ec
change device for whisper
Browse files- builder/fetch_models.py +2 -2
builder/fetch_models.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import se_extractor as se
|
2 |
|
3 |
_ = se.generate_voice_segments('openai_source_output.mp3',vad=False)
|
4 |
-
_ = se.
|
5 |
|
6 |
from resemble_enhance.enhancer.inference import denoise, enhance
|
7 |
import torchaudio
|
@@ -10,4 +10,4 @@ import torchaudio
|
|
10 |
dwav, sr = torchaudio.load('openai_source_output.mp3')
|
11 |
dwav = dwav.mean(dim=0)
|
12 |
|
13 |
-
wav1, new_sr = enhance(dwav, sr, '
|
|
|
1 |
import se_extractor as se
|
2 |
|
3 |
_ = se.generate_voice_segments('openai_source_output.mp3',vad=False)
|
4 |
+
_ = se.load_model()
|
5 |
|
6 |
from resemble_enhance.enhancer.inference import denoise, enhance
|
7 |
import torchaudio
|
|
|
10 |
dwav, sr = torchaudio.load('openai_source_output.mp3')
|
11 |
dwav = dwav.mean(dim=0)
|
12 |
|
13 |
+
wav1, new_sr = enhance(dwav, sr, 'cuda', nfe=32, solver='midpoint', lambd=0.9, tau=0.5)
|