Installation works now by forcing tensorflow version

setup.py

@@ -28,7 +28,7 @@ setup(
         "imageio==2.8.0",
         "omero-py>=5.6.2",
         "zeroc-ice==3.6.5",
-        "tensorflow>=1.14,<=2.3",
+        "tensorflow==2.3",
         "baby@git+ssh://git@git.ecdf.ed.ac.uk/swain-lab/python-pipeline/baby.git@training",
         "logfile_parser@git+ssh://git@git.ecdf.ed.ac.uk/swain-lab/python-pipeline/logfile_parser.git",
         # "extraction@git+ssh://git@git.ecdf.ed.ac.uk/swain-lab/python-pipeline/extraction.git@dev",