fix: IndentationError

Author: raselmandol

train_mnist.py

@@ -99,7 +99,8 @@ def forward(self, x):
 sample_imgs = save_test_samples_images(model, test_loader, device, out_dir="images", num_samples=5)
 test_log += "\n### Sample Images\n"
 for i, (idx, label, pred, img_path) in enumerate(sample_imgs, start=1):
-test_log += "![Test Accuracy](images/test_accuracy.png)\n"
+    img_name = os.path.basename(img_path)
+    test_log += f"![Sample {i:02d}](images/{img_name}) True: {label}, Pred: {pred}\n"
 save_markdown_log("test_output.md", test_log)
 # Mirror test log into docs/
 save_markdown_log(os.path.join("docs", "test_output.md"), test_log)