<reponame>shunkakinoki/slsf-celeb
from fastai.conv_builder import *
from lib.utils import get_labels


def classification_model(arch=resnet34, **kwargs):
	opts = dict(is_multi=False, is_reg=False, pretrained=False)
	n_labels = len(get_labels(os.environ['LABELS_PATH']))
	conv = ConvnetBuilder(arch, n_labels, **opts, **kwargs)
	return conv.model
