Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
|
|
3 |
from transformers import VitsModel, AutoTokenizer,pipeline
|
4 |
import torch
|
5 |
import numpy as np
|
6 |
-
import
|
7 |
|
8 |
|
9 |
def TTS(text):
|
@@ -12,7 +12,7 @@ def TTS(text):
|
|
12 |
inputs = tokenizer(text, return_tensors="pt")
|
13 |
pipe = pipeline("text-to-speech", model=model,tokenizer=tokenizer)
|
14 |
with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
|
15 |
-
|
16 |
return fp.name
|
17 |
iface = gr.Interface(fn=TTS, inputs="text", outputs="audio")
|
18 |
iface.launch(share=False)
|
|
|
3 |
from transformers import VitsModel, AutoTokenizer,pipeline
|
4 |
import torch
|
5 |
import numpy as np
|
6 |
+
import torchaudio
|
7 |
|
8 |
|
9 |
def TTS(text):
|
|
|
12 |
inputs = tokenizer(text, return_tensors="pt")
|
13 |
pipe = pipeline("text-to-speech", model=model,tokenizer=tokenizer)
|
14 |
with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
|
15 |
+
torchaudio.save(fp, pipe(text)['audio'], rate=pipe(text)['sampling_rate'])
|
16 |
return fp.name
|
17 |
iface = gr.Interface(fn=TTS, inputs="text", outputs="audio")
|
18 |
iface.launch(share=False)
|