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)
|