I originally used the PyTorch version but my boss wants me to use the TensorFlow versions so I converted the pytroch one to TensorFlow and exported that from hugging face but unfortunately I am having problems loading the h5  format inside TensorFlow. Well I will try getting help from the TensorFlow team , hope you get the saved model version working eventually