text: from Prognostic.vis.resnet import (resnet18, resnet34, resnet50, resnet101,
                                     resnet152)
  
  MODELS = {'resnet18': resnet18,
            'resnet34': resnet34,
            'resnet50': resnet50,
            'resnet101': resnet101,
            'resnet152': resnet152
            }
  