HI @Ujjwal
Thanks for your detailed feedback
Given the structure of SavedModel, I don’t think this request is address-able in general. TF Hub only can expose elements that a publisher decided to provide callable end-points for.
For the modification that you are looking for working directly with code like the ones from Model Garden 2 ( blog post 1 is the way to go.