From b1eeffa5b590f37e692f074026c48bd9562e7fad Mon Sep 17 00:00:00 2001
From: Yoon Kim <yoon.r.kim@gmail.com>
Date: Fri, 31 Jul 2020 12:44:44 -0400
Subject: [PATCH] Update preprocess.py

---
 preprocess.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/preprocess.py b/preprocess.py
index 3778849..279632e 100755
--- a/preprocess.py
+++ b/preprocess.py
@@ -59,6 +59,7 @@ def prune_vocab(self, k, cnt = False):
 
     def load_vocab(self, vocab_file):
         self.d = {}
+        self.idx2word = {}
         for line in open(vocab_file, 'r'):
             v, k = line.strip().split()
             self.d[v] = int(k)