ASG Models commited on
Commit
a4012f3
·
verified ·
1 Parent(s): 2e90130

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"].to(device)).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
 
 
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