diff --git a/requirements.txt b/requirements.txt
index 4bad5bb..f475316 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 numpy
-setuptools >= 41
+setuptools >= 50
 opencv-python-headless
-ocrd >= 2.22.3
-tensorflow-gpu >= 2.6.0
+ocrd >= 2.23.3
+tensorflow >= 2.6.0
diff --git a/sbb_binarize/sbb_binarize.py b/sbb_binarize/sbb_binarize.py
index f7f7ea9..b754b44 100644
--- a/sbb_binarize/sbb_binarize.py
+++ b/sbb_binarize/sbb_binarize.py
@@ -91,7 +91,8 @@ class SbbBinarizer:
 
         self.start_new_session()
 
-        self.model_files = glob('%s/*.h5' % self.model_dir)
+        #self.model_files = glob('%s/*.h5' % self.model_dir)
+        self.model_files = glob(self.model_dir+"/*/", recursive = True)
 
         self.models = []
         for model_file in self.model_files: