import os.path

from .model import F3Classification
from .const import CLASS_CN_LIST, CLASS_OTHER_FIRST

classifier = F3Classification(
    class_name_list=CLASS_CN_LIST,
    class_other_first=CLASS_OTHER_FIRST
)

classifier.load_model(load_weights_path=os.path.join(
    os.path.dirname(os.path.abspath(__file__)), 'ckpt_prod.h5'))