code: def get_model_loader(filename):
      
      assert isinstance(filename, six.string_types), filename
      filename = os.path.expanduser(filename)
      if filename.endswith('.npy'):
          assert tf.gfile.Exists(filename), filename
          return DictRestore(np.load(filename, encoding='latin1').item())
      elif filename.endswith('.npz'):
          assert tf.gfile.Exists(filename), filename
          obj = np.load(filename)
          return DictRestore(dict(obj))
      else:
          return SaverRestore(filename)
docs: Get a corresponding model loader by looking at the file name.
  
      Returns:
          SessInit: either a :class:`DictRestore` (if name ends with 'npy/npz') or
          :class:`SaverRestore` (otherwise).