github/obs-localvocal
1
1
Fork 0
mirror of https://github.com/royshil/obs-localvocal.git synced 2026-01-10 04:48:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Add regex check to detect false prediction in whisper-processing.cpp

Browse Source
This commit is contained in:
Roy Shilkrot
2024-06-03 00:34:38 -04:00
parent a6529a41d2
commit dfbacfff82
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/whisper-utils/whisper-processing.cpp
View File

@@ -18,6 +18,7 @@
#include <algorithm>
#include <chrono>
#include <regex>
struct vad_state {
bool vad_on;
@@ -275,6 +276,13 @@ struct DetectionResultWithText run_whisper_inference(struct transcription_filter
return {DETECTION_RESULT_SILENCE, "", t0, t1, {}};
}
// Check regex for "MBC .*" to detect false prediction
std::regex mbc_regex("MBC.*");
if (std::regex_match(text, mbc_regex)) {
obs_log(gf->log_level, "False prediction detected: %s", text.c_str());
return {DETECTION_RESULT_SILENCE, "", t0, t1, {}};
}
return {DETECTION_RESULT_SPEECH, text, t0, t1, tokens};
}
}