INNER CODE UNIT · Python

__call__

graviraja/100-Days-of-NLP · applications/generation/image_captioning/latex_app/app.py:50

    def __call__(self, sign):
        if not sign in self.sign2id:
            return self.sign2id['<unk>']
        return self.sign2id[sign]

def load_vocab():
    with open(os.path.join("model", 'vocab.pkl'), 'rb') as f:
        vocab = pickle.load(f)
    return vocab

@st.cache
def load_model():
    embed_dim = 256
    encoder_dim = 512
    decoder_dim = 512
    attention_dim = 512
    dropout = 0.3
    vocab = load_vocab()

View source record →