infer_tf

import tensorflow_addons as tfa
import tensorflow as tf

tfa.register_all()
dir_saved_model = '/home/pikaliov/Projects/deeppunctrestorejustencoder/models_and_weights/transformer_asr_restore/model_dir/models_1_21.15_0.78/'
loaded = tf.saved_model.load(dir_saved_model)
infer = loaded.signatures["serving_default"]


rand_val = tf.random.uniform(shape=[1, 100], maxval=15, dtype=tf.int32)
res = infer(rand_val)['output_1']
print(res)