雾聪
2023-09-01 beceb14feca0befb9a0f46a9fe3350546874f867
funasr/runtime/onnxruntime/src/funasrruntime.cpp
@@ -224,11 +224,17 @@
         return nullptr;
      funasr::Audio audio(1);
      try{
      if(wav_format == "pcm" || wav_format == "PCM"){
         if (!audio.LoadPcmwav(sz_buf, n_len, &sampling_rate))
            return nullptr;
      }else{
         if (!audio.FfmpegLoad(sz_buf, n_len))
               return nullptr;
         }
      }catch (std::exception const &e)
      {
         LOG(ERROR)<<e.what();
            return nullptr;
      }
@@ -288,6 +294,7 @@
         return nullptr;
      
      funasr::Audio audio(1);
      try{
      if(funasr::is_target_file(sz_filename, "wav")){
         int32_t sampling_rate_ = -1;
         if(!audio.LoadWav(sz_filename, &sampling_rate_))
@@ -299,6 +306,12 @@
         if (!audio.FfmpegLoad(sz_filename))
            return nullptr;
      }
      }catch (std::exception const &e)
      {
         LOG(ERROR)<<e.what();
         return nullptr;
      }
      funasr::FUNASR_RECOG_RESULT* p_result = new funasr::FUNASR_RECOG_RESULT;
      p_result->snippet_time = audio.GetTimeLen();
      if(p_result->snippet_time == 0){