Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -101,7 +101,7 @@ def predict(text, speaker):
|
|
| 101 |
# input_ids = input_ids[..., :model.config.max_text_positions]
|
| 102 |
|
| 103 |
### ### ###
|
| 104 |
-
example = dataset['train'][
|
| 105 |
speaker_embeddings = torch.tensor(example["speaker_embeddings"]).unsqueeze(0)
|
| 106 |
|
| 107 |
spectrogram = model.generate_speech(inputs["input_ids"], speaker_embeddings)
|
|
|
|
| 101 |
# input_ids = input_ids[..., :model.config.max_text_positions]
|
| 102 |
|
| 103 |
### ### ###
|
| 104 |
+
example = dataset['train'][888]
|
| 105 |
speaker_embeddings = torch.tensor(example["speaker_embeddings"]).unsqueeze(0)
|
| 106 |
|
| 107 |
spectrogram = model.generate_speech(inputs["input_ids"], speaker_embeddings)
|