divakaivan commited on
Commit
40de861
1 Parent(s): 97da773

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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'][825]
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)