Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|