Spaces:
Sleeping
Sleeping
add None option for speaker
Browse files
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)
|