fill missing argument in argmax: axis

main.cpp

@@ -61,7 +61,7 @@ int main() {
         std::cout << "  Loading model...\n";
         InferenceResult result = run_inference(processed);
         std::cout << "  Inference succeeded!\n";
-	std::cout << nc::argmax(result.logits);   // seems wrong
+	std::cout << nc::argmax(result.logits, nc::Axis::ROW);   // something is weird...
         // Cleanup
         stbi_image_free(pixels);
         free(ppm.pixels);