owaski commited on
Commit
8d8f4e3
·
1 Parent(s): 4b89453

add None option for speaker

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -144,7 +144,7 @@ def predict(audio, src_language, tgt_language, mic_audio=None):
144
  'prev_output_tokens': None,
145
  },
146
  'target_lengths': None,
147
- 'speaker': tts_sample['speaker'].to(device)
148
  }
149
 
150
  wav, rate = TTSHubInterface.get_prediction(tts_task, tts_model, tts_generator, tts_sample)
 
144
  'prev_output_tokens': None,
145
  },
146
  'target_lengths': None,
147
+ 'speaker': tts_sample['speaker'].to(device) if tts_sample['speaker'] is not None else None,
148
  }
149
 
150
  wav, rate = TTSHubInterface.get_prediction(tts_task, tts_model, tts_generator, tts_sample)