Update README.md

README.md

@@ -1,18 +1,28 @@
+---
+license: apache-2.0
+datasets:
+- mnist
+language:
+- en
+metrics:
+- accuracy
+library_name: keras
+---
 # 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. 
 
 The model will be used as a preliminary step in solving Sudokus through linear programming. Model It is responsible for classifying the content of each sudoku cell:
 
-class_0: empty cell
-class_1: cell contains the number 1
-class_2: cell contains the number 2
-class_3: cell contains the number 3
-class_4: cell contains the number 4
-class_5: cell contains the number 5
-class_6: cell contains the number 6
-class_7: cell contains the number 7
-class_8: cell contains the number 8
-class_9: cell contains the number 9
+- class_0: empty cell
+- class_1: cell contains the number 1
+- class_2: cell contains the number 2
+- class_3: cell contains the number 3
+- class_4: cell contains the number 4
+- class_5: cell contains the number 5
+- class_6: cell contains the number 6
+- class_7: cell contains the number 7
+- class_8: cell contains the number 8
+- 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.
 
@@ -35,6 +45,4 @@ An initial validation is performed. It remains pending to increase the size of t
 The results of the inference are as follows:
 <div style="text-align: center;">
 <img src="https://i.imgur.com/U2MJzH6.jpg" width="500">
-</div>
-
-
+</div>