reach-vb HF Staff commited on
Commit
f867e19
·
1 Parent(s): 0ee0e76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,13 +20,13 @@ speaker_embeddings = {
20
  }
21
 
22
 
23
- def predict(text, speaker):
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)