unfortunately, its unclear, what you did there, or even what they are using.
impossible to debug
as of now, there is also classification only, right ?
importing tf2 saved_model is not fully supported yet
in the end, rather use something tried and tested