Update examples/llava/mtmd-cli.cpp

Co-authored-by: Sigbjørn Skjæret <[email protected]>

Author: ahmedshakill

examples/llava/mtmd-cli.cpp

@@ -308,7 +308,11 @@ int main(int argc, char ** argv) {
             }
             g_is_generating = true;
             if (line.find("/image") == 0) {
-                std::string image = (line.size()>=7) ? line.substr(7) : "";
+                if (line.size() < 8) {
+                    LOG_ERR("ERR: Missing image filename\n");
+                    continue;
+                }
+                std::string image = line.substr(7);
                 images_fname.push_back(string_strip(image));
                 content += "<__image__>";
                 continue;