stuff

app.py

@@ -47,6 +47,7 @@ def respond(input):
 
         out = model.out(model.decoder(outputs[:i].unsqueeze(0), e_outputs, src_mask, trg_mask))
         out = torch.nn.functional.softmax(out, dim=-1).detach()
+        print(out.shape)
         print(out[:, -1].data[0])
         print(out[:, -1].data[0].shape)
         print(np.sum(out[:, -1].data[0]))