From 81a96f73c8df413625d07bf8c2ff5d5f2ea88359 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Thu, 28 May 2020 23:16:51 -0400 Subject: [PATCH] better error messages --- tts_service/src/endpoints/speak.rs | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tts_service/src/endpoints/speak.rs b/tts_service/src/endpoints/speak.rs index 38d16424c5..041cdad2c7 100644 --- a/tts_service/src/endpoints/speak.rs +++ b/tts_service/src/endpoints/speak.rs @@ -63,8 +63,11 @@ pub async fn post_speak(_request: HttpRequest, println!("Encoded Text: {:?}", encoded); - let tacotron = app_state.model_cache.get_or_load_arbabet_tacotron(&tacotron_model).unwrap(); - let melgan = app_state.model_cache.get_or_load_melgan(&melgan_model).unwrap(); + let tacotron = app_state.model_cache.get_or_load_arbabet_tacotron(&tacotron_model) + .expect(&format!("Couldn't load tacotron model: {}", tacotron_model)); + + let melgan = app_state.model_cache.get_or_load_melgan(&melgan_model) + .expect(&format!("Couldn't load melgan model: {}", melgan_model)); let wav_data = TacoMelModel::new().run_tts_encoded(&tacotron, &melgan, &encoded);