Spaces:
Sleeping
Sleeping
Irpan
commited on
Commit
·
8377a77
1
Parent(s):
30e5da4
asr
Browse files- requirements.txt +3 -1
- tts.py +1 -1
requirements.txt
CHANGED
@@ -2,4 +2,6 @@ gradio==2.9.1
|
|
2 |
torch
|
3 |
torchaudio
|
4 |
transformers
|
5 |
-
numpy
|
|
|
|
|
|
2 |
torch
|
3 |
torchaudio
|
4 |
transformers
|
5 |
+
numpy
|
6 |
+
scipy==1.13.1
|
7 |
+
umsc
|
tts.py
CHANGED
@@ -17,7 +17,7 @@ def synthesize(text, model_id):
|
|
17 |
inputs = processor(text, return_tensors="pt").to(device)
|
18 |
|
19 |
with torch.no_grad():
|
20 |
-
output =
|
21 |
|
22 |
output_path = "tts_output.wav"
|
23 |
sample_rate = 16000
|
|
|
17 |
inputs = processor(text, return_tensors="pt").to(device)
|
18 |
|
19 |
with torch.no_grad():
|
20 |
+
output = model(**inputs).waveform.cpu() # Move output back to CPU for saving
|
21 |
|
22 |
output_path = "tts_output.wav"
|
23 |
sample_rate = 16000
|