Update app.py
Browse files
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
|