Update README.md

README.md

@@ -11,7 +11,7 @@ with just under 50k parameters, this model should run on any hardware.
 ## List of included files
 train.py: the training function of the model, can be used to train a new model on different data.
 
-main.h5: the model trained on the data.csv file for 3 epoches with a batch size of 1. generorated by train.py
+main.h5: the model trained on the data.csv file for 2 epoches with a batch size of 1. generorated by train.py
 
 tokenizer.pkl: contain the tokenizer for 5he pre-trained model. genorated by train.py