Stopwolf commited on
Commit
c022e43
1 Parent(s): 9a22fd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -30,7 +30,8 @@ def synthesise(text):
30
  inputs = tokenizer(text=text, return_tensors="pt")
31
 
32
  with torch.no_grad():
33
- output = model(**inputs).waveform
 
34
  # inputs = processor(text=text, return_tensors="pt")
35
  # speech = model.generate_speech(inputs["input_ids"].to(device), speaker_embeddings.to(device), vocoder=vocoder)
36
  return output.cpu()
 
30
  inputs = tokenizer(text=text, return_tensors="pt")
31
 
32
  with torch.no_grad():
33
+ output = model(**inputs)
34
+ output = output.waveform
35
  # inputs = processor(text=text, return_tensors="pt")
36
  # speech = model.generate_speech(inputs["input_ids"].to(device), speaker_embeddings.to(device), vocoder=vocoder)
37
  return output.cpu()