mirrors/obs-localvocal
1
0
Fork 0
mirror of https://github.com/occ-ai/obs-localvocal synced 2024-11-07 18:57:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
Roy Shilkrot 2023-10-29 01:45:50 -04:00
parent 030bb819c3
commit ad182f4593
1 changed files with 19 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/whisper-processing.cpp
View File

@ -77,25 +77,27 @@ struct whisper_context *init_whisper_context(const std::string &model_path)
obs_log(LOG_INFO, "Loading whisper model from %s", model_path.c_str());
#ifdef _WIN32
// convert model path UTF8 to wstring (wchar_t) for whisper
int count = MultiByteToWideChar(CP_UTF8, 0, model_path.c_str(), (int)model_path.length(), NULL, 0);
std::wstring model_path_ws(count, 0);
MultiByteToWideChar(CP_UTF8, 0, model_path.c_str(), (int)model_path.length(), &model_path_ws[0], count);
// convert model path UTF8 to wstring (wchar_t) for whisper
int count = MultiByteToWideChar(CP_UTF8, 0, model_path.c_str(), (int)model_path.length(),
NULL, 0);
std::wstring model_path_ws(count, 0);
MultiByteToWideChar(CP_UTF8, 0, model_path.c_str(), (int)model_path.length(),
&model_path_ws[0], count);
// Read model into buffer
std::ifstream modelFile(model_path_ws, std::ios::binary);
if (!modelFile.is_open()) {
obs_log(LOG_ERROR, "Failed to open whisper model file %s", model_path.c_str());
return nullptr;
}
modelFile.seekg(0, std::ios::end);
const size_t modelFileSize = modelFile.tellg();
modelFile.seekg(0, std::ios::beg);
std::vector<char> modelBuffer(modelFileSize);
modelFile.read(modelBuffer.data(), modelFileSize);
modelFile.close();
// Read model into buffer
std::ifstream modelFile(model_path_ws, std::ios::binary);
if (!modelFile.is_open()) {
obs_log(LOG_ERROR, "Failed to open whisper model file %s", model_path.c_str());
return nullptr;
}
modelFile.seekg(0, std::ios::end);
const size_t modelFileSize = modelFile.tellg();
modelFile.seekg(0, std::ios::beg);
std::vector<char> modelBuffer(modelFileSize);
modelFile.read(modelBuffer.data(), modelFileSize);
modelFile.close();
// Initialize whisper
// Initialize whisper
struct whisper_context *ctx = whisper_init_from_buffer(modelBuffer.data(), modelFileSize);
#else
struct whisper_context *ctx = whisper_init_from_file(model_path.c_str());