Return timestamp info and tokens in offline ASR

Co-authored-by: zhangbaofeng@npnets.com <41259@Zbf>
This commit is contained in:
cooldoomsday
2023-05-06 10:20:46 +08:00
committed by GitHub
parent 8ce6fff9b6
commit 0bc571f6ee
4 changed files with 14 additions and 4 deletions

View File

@@ -8,7 +8,7 @@
#include <algorithm>
#include <cmath>
#include "nlohmann/json.hpp"
#include "kaldi-native-fbank/csrc/online-feature.h"
#include "sherpa-onnx/csrc/macros.h"
#include "sherpa-onnx/csrc/offline-recognizer.h"
@@ -214,5 +214,12 @@ void OfflineStream::SetResult(const OfflineRecognitionResult &r) {
const OfflineRecognitionResult &OfflineStream::GetResult() const {
return impl_->GetResult();
}
std::string OfflineRecognitionResult::AsJsonString() const {
nlohmann::json j;
j["text"] = text;
j["tokens"] = tokens;
j["timestamps"] = timestamps;
return j.dump();
}
} // namespace sherpa_onnx