add logs for model load

雾聪

2023-06-16 4382dcb15fe5e1ec0a03c0852998a4000ccf7161

add logs for model load

 funasr/runtime/onnxruntime/src/ct-transformer.cpp

@@ -18,6 +18,7 @@

    try{
        m_session = std::make_unique<Ort::Session>(env_, punc_model.c_str(), session_options);
        LOG(INFO) << "Successfully load model from " << punc_model;
    }
    catch (std::exception const &e) {
        LOG(ERROR) << "Error when load punc onnx model: " << e.what();

 funasr/runtime/onnxruntime/src/paraformer.cpp

@@ -33,6 +33,7 @@

    try {
        m_session = std::make_unique<Ort::Session>(env_, am_model.c_str(), session_options);
        LOG(INFO) << "Successfully load model from " << am_model;
    } catch (std::exception const &e) {
        LOG(ERROR) << "Error when load am onnx model: " << e.what();
        exit(0);

 funasr/runtime/websocket/funasr-wss-client.cpp

@@ -217,7 +217,7 @@
                }
            }
            if (wait) {
                LOG(INFO) << "wait.." << m_open;
                // LOG(INFO) << "wait.." << m_open;
                WaitABit();
                continue;
            }
@@ -291,7 +291,7 @@
                                       false, 1, "int");
    TCLAP::ValueArg<int> is_ssl_(
        "", "is-ssl", "is-ssl is 1 means use wss connection, or use ws connection", 
        false, 0, "int");
        false, 1, "int");

    cmd.add(server_ip_);
    cmd.add(port_);