Edmon02 commited on
Commit
4ef5dcd
1 Parent(s): 51dcd32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -155,7 +155,7 @@ def predict(text, speaker):
155
  input_ids = inputs["input_ids"]
156
  input_ids = input_ids[..., :model.config.max_text_positions]
157
 
158
- speaker_embedding = np.load(speaker_embeddings[speaker[:3]])
159
 
160
  speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
161
 
 
155
  input_ids = inputs["input_ids"]
156
  input_ids = input_ids[..., :model.config.max_text_positions]
157
 
158
+ speaker_embedding = np.load(speaker_embeddings[speaker[:3]]).astype(np.float32)
159
 
160
  speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
161