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