diff --git a/model/build_datasets.py b/model/build_datasets.py
index 513d883..d32138d 100755
--- a/model/build_datasets.py
+++ b/model/build_datasets.py
@@ -60,7 +60,7 @@
         content_hash = chars.replace(" ", "").replace("\n", "")
         content_hash = hashlib.sha256(content_hash.encode('utf-8')).hexdigest()
 
-        if len(eval_set) == evaluation_samples_number:
+        if len(eval_set) >= evaluation_samples_number:
             train_set.append(path)
         else:
             is_unique = True
@@ -76,8 +76,6 @@
 
         hashes.append(content_hash)
 
-assert len(eval_set) == evaluation_samples_number
-assert len(train_set) == training_samples_number
 
 if not os.path.exists("{}/{}".format(os.path.dirname(input_path), EVALUATION_SET_NAME)):
     os.makedirs("{}/{}".format(os.path.dirname(input_path), EVALUATION_SET_NAME))
diff --git a/model/train.py b/model/train.py
index afa4424..8eb6bda 100755
--- a/model/train.py
+++ b/model/train.py
@@ -3,8 +3,6 @@
 from __future__ import absolute_import
 __author__ = 'Tony Beltramelli - www.tonybeltramelli.com'
 
-import tensorflow as tf
-sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
 
 import sys
 
diff --git a/requirements.txt b/requirements.txt
index 3071c5e..88a88cd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
-Keras==2.1.2
-numpy==1.13.3
-opencv-python==3.3.0.10
-h5py==2.7.1
-tensorflow==1.4.0
+Keras==2.13.1
+numpy==1.23.5
+opencv-python==4.9.0.80
+h5py==3.10.0
+tensorflow==2.2