from musicfm.models.musicfm_fma import MusicFM_FMA
from musicfm.models.musicfm_mertlong import MusicFM_MERTLong
from musicfm.models.musicfm_7q import MusicFM7q
from musicfm.models.musicfm_dual import MusicFM_Dual
from musicfm.models.musicfm_100hz import MusicFM100Hz


def get_model(model_type):
    if model_type == "mertlong":
        return MusicFM_MERTLong

    elif model_type == "fma":
        return MusicFM_FMA

    elif model_type == "7q":
        return MusicFM7q

    elif model_type == "dual":
        return MusicFM_Dual

    elif model_type == "100hz":
        return MusicFM100Hz

    else:
        print("%s model type is not supported yet" % model_type)