diff --git a/examples/server/server.cpp b/examples/server/server.cpp
index 5e4dffadf39c5..2f41f45677c72 100644
--- a/examples/server/server.cpp
+++ b/examples/server/server.cpp
@@ -2592,10 +2592,15 @@ int main(int argc, char ** argv) {
return false;
};
- auto middleware_server_state = [&res_error, &state](const httplib::Request &, httplib::Response & res) {
+ auto middleware_server_state = [&res_error, &state](const httplib::Request & req, httplib::Response & res) {
server_state current_state = state.load();
if (current_state == SERVER_STATE_LOADING_MODEL) {
- res_error(res, format_error_response("Loading model", ERROR_TYPE_UNAVAILABLE));
+ if(req.path == "/"){
+ res.set_content("The model is loading. Please wait.
The user interface will appear soon.
You may need to refresh the page.", "text/html; charset=utf-8");
+ res.status = 503;
+ } else {
+ res_error(res, format_error_response("Loading model", ERROR_TYPE_UNAVAILABLE));
+ }
return false;
}
return true;