INNER CODE UNIT · Python
__len__
graviraja/100-Days-of-NLP · applications/generation/image_captioning/latex_app/app.py:47
def __len__(self):
return self.length
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