Spaces:
Sleeping
Sleeping
ASG Models
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -91,7 +91,7 @@ def modelspeech(text):
|
|
91 |
with torch.no_grad():
|
92 |
inputs = tokenizer(text, return_tensors="pt")#.cuda()
|
93 |
|
94 |
-
wav = model(input_ids=inputs["input_ids"]
|
95 |
# display(Audio(wav, rate=model.config.sampling_rate))
|
96 |
return model.config.sampling_rate,wav#remove_noise_nr(wav)
|
97 |
|
|
|
91 |
with torch.no_grad():
|
92 |
inputs = tokenizer(text, return_tensors="pt")#.cuda()
|
93 |
|
94 |
+
wav = model(input_ids=inputs["input_ids"]).waveform.cpu().numpy().reshape(-1)
|
95 |
# display(Audio(wav, rate=model.config.sampling_rate))
|
96 |
return model.config.sampling_rate,wav#remove_noise_nr(wav)
|
97 |
|