Update README.md

README.md

@@ -1,6 +1,3 @@
----
-license: apache-2.0
----
 # Description
 This model is a convolutional neural network built with transfer learning using the pre-trained model 'VGG16.' The 'block5_conv1' layer is retrained, and a final dense layer with 128 neurons is added. 
 
@@ -20,7 +17,7 @@ class_9: cell contains the number 9
 The dataset is constructed with balanced classes using images from the famous "MNIST digits classification" dataset, as well as images of numbers written digitally.
 
 # Dataset schema
-After applying data augmentation to the dataset, the total number of images is as follows:
+The image size it is 28x28. After applying data augmentation to the dataset, the total number of images is as follows:
 - Training images: 5600
 - Validation images: 2400
 - Test images: 200
@@ -30,10 +27,14 @@ Test Accuracy: 0.9810
 # Other validations:
 An initial validation is performed. It remains pending to increase the size of the validations to understand the reliability of the mode
 
-![Sudoku example](https://imgur.com/gallery/kdj9udt)
+<div style="text-align: center;">
+<img src="https://i.imgur.com/kdj9udt.jpg" width="300">
+</div>
+</div>
 
 The results of the inference are as follows:
-
-![Confusion Matrix](https://imgur.com/gallery/U2MJzH6)
+<div style="text-align: center;">
+<img src="https://i.imgur.com/U2MJzH6.jpg" width="500">
+</div>