ChuGyouk commited on
Commit
2ceb926
1 Parent(s): a07ae39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def synthesise(text):
29
  inputs = processor(text=text, return_tensors="pt")
30
  outputs = model(input_ids=inputs["input_ids"].to(device), speaker_id=speaker_embeddings.to(device))
31
  speech = outputs["waveform"]
32
- return speech.detach().cpu()
33
 
34
 
35
  target_dtype = np.int16
 
29
  inputs = processor(text=text, return_tensors="pt")
30
  outputs = model(input_ids=inputs["input_ids"].to(device), speaker_id=speaker_embeddings.to(device))
31
  speech = outputs["waveform"]
32
+ return speech.squeeze(0).detach().cpu()
33
 
34
 
35
  target_dtype = np.int16