Update utils.py

utils.py

@@ -61,11 +61,11 @@ def generate_text(seed, model, vocab, max_len=20, temperature=1., device=device,
     idxs = []
     for _ in range(max_len):
         yhat = model(x)
-        prob = yhat[:, -1].softmax(dim=-1).squeeze()
-        prob /=  temperature
-        top_probs = torch.topk(prob, top_k, dim=-1).indices
-        prob[~top_probs] = 0.
-        idx = torch.multinomial(prob, 1, replacement=True).item()
+        probs = yhat[:, -1].softmax(dim=-1).squeeze()
+        probs /=  temperature
+        top_probs = torch.topk(probs, top_k, dim=-1).indices
+        probs[~top_probs] = 0.
+        idx = torch.multinomial(probs, 1, replacement=True).item()
         idxs.append(idx)
         x = torch.cat([x, torch.ones(1, 1).fill_(idx).long().to(device)], dim=1)
         if itos[idx] == '<eos>':