Prediction.py

start = np.random.randint(0,len(valid_list) - 150)
seed_list = valid_list[start:start+20]

current_string = [string2int[string] for string in seed_list]
for i in range(20):
  input = torch.tensor(current_string,dtype=torch.long).cuda().reshape(1,-1)
  output = model(input)
  _, predicted = torch.max(output[:,-1,:], 1)
  current_string.append(predicted.item())
print(' '.join(seed_list))
print(' '.join(int2string[x] for x in current_string))