Skip to content

Commit

Permalink
Update offline-recognizer-whisper-impl.h
Browse files Browse the repository at this point in the history
  • Loading branch information
iprovalo committed Feb 21, 2025
1 parent 6e83e67 commit 3f319a2
Showing 1 changed file with 25 additions and 25 deletions.
50 changes: 25 additions & 25 deletions sherpa-onnx/csrc/offline-recognizer-whisper-impl.h
Original file line number Diff line number Diff line change
Expand Up @@ -24,31 +24,6 @@
namespace sherpa_onnx {

class OfflineRecognizerWhisperImpl : public OfflineRecognizerImpl {
private:
OfflineRecognitionResult Convert(const OfflineWhisperDecoderResult &src,
const SymbolTable &sym_table) const {
OfflineRecognitionResult r;
r.tokens.reserve(src.tokens.size());

std::string text;
for (auto i : src.tokens) {
if (!sym_table.Contains(i)) {
continue;
}

std::string s = sym_table[i];
s = ApplyInverseTextNormalization(s);

text += s;
r.tokens.push_back(s);
}

r.text = text;
r.lang = src.lang;

return r;
}

public:
explicit OfflineRecognizerWhisperImpl(const OfflineRecognizerConfig &config)
: OfflineRecognizerImpl(config),
Expand Down Expand Up @@ -171,6 +146,31 @@ class OfflineRecognizerWhisperImpl : public OfflineRecognizerImpl {
}
}

private:
OfflineRecognitionResult Convert(const OfflineWhisperDecoderResult &src,
const SymbolTable &sym_table) const {
OfflineRecognitionResult r;
r.tokens.reserve(src.tokens.size());

std::string text;
for (auto i : src.tokens) {
if (!sym_table.Contains(i)) {
continue;
}

std::string s = sym_table[i];
s = ApplyInverseTextNormalization(s);

text += s;
r.tokens.push_back(s);
}

r.text = text;
r.lang = src.lang;

return r;
}

private:
OfflineRecognizerConfig config_;
SymbolTable symbol_table_;
Expand Down

0 comments on commit 3f319a2

Please sign in to comment.