How to export the onsets_frames_transcription checkpoint to a pb model for TFServing?

Question

I want to use onsets_frames_transcription for serving, but the preprocessing of the audio example proto is related to the data.provide_batch, and it returns a dataset object