RuntimeError: Failed. Model function mildnet_vgg16 not found

[noptanit12]

I try to run your model !bash gcloud.local.run.keras.sh $MILDNET_CONFIG on my Jupyter but it has the RuntimeError: Failed. Model function mildnet_vgg16 not found

please kindly help to solve this problem.

Best Regards,

More details;
job_configs/mildnet.cnf
Python 2.7.16
TF 2.3.0

in task.py at line 8,9, I have to commented
#from tensorflow import set_random_seed
#set_random_seed(2)
and changed to the
import tensorflow
tensorflow.random.set_seed(2)

and add @tf.function before def main