9f200d963b9646fb50326f7cf6788021ddfe955d,kraken/lib/models.py,,load_any,#,172

Before Change

                    seq = load_pyrnn(fname)
                    seq.kind = "pyrnn"
                    return seq
                except Exception as e:
                    raise
            else:
                raise

def load_clstm(fname):

After Change

    nn = None
    kind = ""
    fname = abspath(expandvars(expanduser(fname)))
    try:
        nn = TorchVGSLModel.load_model(fname)
        kind = "vgsl"
    except:
        nn = TorchVGSLModel.load_clstm_model(fname)
        kind = "clstm"
    if not nn:
        raise KrakenInvalidModelException("File {} not loadable by any parser.".format(fname))
    seq = TorchSeqRecognizer(nn)
    seq.kind = kind

Instances