thanks for the code, I tried running that which got some issue. I tried conda , python 3.6 and tensorflow 2.0 and 1.15, not give me good. Can you please add a requirement.txt file for the environment used for this code.