Update README.md

README.md

@@ -18,7 +18,7 @@ datasets:
 
 ## How the Model Was Trained
 
-The model was trained using the MNIST dataset, which consists of 60,000 training images and 10,000 test images. Each image is a 28x28 grayscale representation of a handwritten digit. Training was conducted over 20 epochs with a batch size of 32. The SGD optimizer was used with a learning rate of 0.01.
+The model was trained using the MNIST dataset, which consists of 60,000 training images and 10,000 test images. Each image is a 28x28 grayscale representation of a handwritten digit. Training was conducted over 32 epochs with a batch size of 32. The SGD optimizer was used with a learning rate of 0.01.
 
 ### Training Script
 
@@ -72,7 +72,20 @@ model.eval()
 
 ## Performance
 
-Describe your model's performance on the test data or during validation. You might want to include metrics such as accuracy, precision, recall, and F1 score.
+<div style="display: flex; justify-content: space-between;">
+    <img src="img/output.png" alt="Training Plot 1" style="width: 45%;">
+    <img src="img/output2.png" alt="Training Plot 2" style="width: 45%;">
+</div>
+
+## Testing Model
+
+<div style="display: grid; grid-template-rows: auto auto; grid-template-columns: 50% 50%; gap: 20px;">
+    <img src="img/myplot1.png" alt="Training Plot 1" style="width: 100%;">
+    <div style="display: grid; grid-template-columns: 50% 50%;">
+        <img src="img/myplot2.png" alt="Training Plot 2" style="width: 100%;">
+        <img src="img/myplot3.png" alt="Training Plot 3" style="width: 100%;">
+    </div>
+</div>
 
 ## Limitations and Ethics