Update app.py
Browse files
app.py
CHANGED
|
@@ -20,13 +20,13 @@ speaker_embeddings = {
|
|
| 20 |
}
|
| 21 |
|
| 22 |
|
| 23 |
-
def predict(text
|
| 24 |
if len(text.strip()) == 0:
|
| 25 |
return (16000, np.zeros(0).astype(np.int16))
|
| 26 |
|
| 27 |
inputs = processor(text=text, return_tensors="pt")
|
| 28 |
|
| 29 |
-
speaker_embedding = np.random.default_rng().uniform(-0.20,0.20,512)
|
| 30 |
speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
|
| 31 |
|
| 32 |
speech = model.generate_speech(inputs["input_ids"], speaker_embedding, vocoder=vocoder)
|
|
|
|
| 20 |
}
|
| 21 |
|
| 22 |
|
| 23 |
+
def predict(text):
|
| 24 |
if len(text.strip()) == 0:
|
| 25 |
return (16000, np.zeros(0).astype(np.int16))
|
| 26 |
|
| 27 |
inputs = processor(text=text, return_tensors="pt")
|
| 28 |
|
| 29 |
+
speaker_embedding = np.float32(np.random.default_rng().uniform(-0.20,0.20,512))
|
| 30 |
speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
|
| 31 |
|
| 32 |
speech = model.generate_speech(inputs["input_ids"], speaker_embedding, vocoder=vocoder)
|