Update README.md

README.md

@@ -1,6 +1,6 @@
----
-pipeline_tag: image-classification
----
+---
+pipeline_tag: image-classification
+---
 ## Table cell classification
 
 The model is trained to classify table cell images as either empty or not empty. It has been trained using 
@@ -29,7 +29,9 @@ Validation dataset consisted of
 ## Training procedure
 
 The code used for model training is available in the repository in `train.py` file, which uses functions from 
-`augment.py` and `utils.py` files. The model was trained using cpu with the following hyperparameters:
+`augment.py` and `utils.py` files. The required libraries are listed in the `requirements.txt` file.
+
+The model was trained using cpu with the following hyperparameters:
 
 - image size: 2560
 - learning rate: 0.0001