diff options
Diffstat (limited to 'src/command/marian_server.cpp')
-rw-r--r-- | src/command/marian_server.cpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/command/marian_server.cpp b/src/command/marian_server.cpp index 2c364940..467b564f 100644 --- a/src/command/marian_server.cpp +++ b/src/command/marian_server.cpp @@ -14,6 +14,7 @@ int main(int argc, char **argv) { // Initialize translation task auto options = parseOptions(argc, argv, cli::mode::server, true); auto task = New<TranslateService<BeamSearch>>(options); + auto quiet = options->get<bool>("quiet-translation"); // Initialize web server WSServer server; @@ -21,8 +22,8 @@ int main(int argc, char **argv) { auto &translate = server.endpoint["^/translate/?$"]; - translate.on_message = [&task](Ptr<WSServer::Connection> connection, - Ptr<WSServer::InMessage> message) { + translate.on_message = [&task, quiet](Ptr<WSServer::Connection> connection, + Ptr<WSServer::InMessage> message) { // Get input text auto inputText = message->string(); auto sendStream = std::make_shared<WSServer::OutMessage>(); @@ -30,9 +31,9 @@ int main(int argc, char **argv) { // Translate timer::Timer timer; auto outputText = task->run(inputText); - LOG(info, "Best translation: {}", outputText); *sendStream << outputText << std::endl; - LOG(info, "Translation took: {:.5f}s", timer.elapsed()); + if(!quiet) + LOG(info, "Translation took: {:.5f}s", timer.elapsed()); // Send translation back connection->send(sendStream, [](const SimpleWeb::error_code &ec) { |