Update app.py

app.py

@@ -60,7 +60,7 @@ padding_idx = vocab['<PAD>']
 model = Transformer(height,width,n_channels,patch_size,dim,enc_head,enc_feed_forward,enc_depth,
                     dec_head,dec_feed_forward,dec_depth,max_seq_len,vocab_size,padding_idx)
 state = torch.load("captioning.ptrom",map_location="cpu")
-model.load_state_dict(state)
+model.load_state_dict(state['model_state_dict'])
 model = model.to(device)
 itos = {v: k for k, v in vocab.items()}