it might be extremely useful to be able to pass a TF model directly, in addition of the keras and torch model class