Update pages/imdb.py

pages/imdb.py

@@ -14,13 +14,6 @@ from data.rnn_preprocessing import (
     preprocess_single_string
 )
 
-# Load LSTM model
-embedding_layer32 = nn.Embedding.from_pretrained(torch.FloatTensor(embedding_matrix))
-VOCAB_SIZE = len(vocab_to_int) + 1  # add 1 for the padding token
-HIDDEN_DIM = 64
-SEQ_LEN = 32
-
-
 # Load Word2Vec model
 wv = Word2Vec.load('models/word2vec32.model')
 embedding_matrix = np.zeros((VOCAB_SIZE, EMBEDDING_DIM))
@@ -32,6 +25,17 @@ for word, i in vocab_to_int.items():
     except KeyError:
         pass
 
+
+
+# Load LSTM model
+embedding_layer32 = nn.Embedding.from_pretrained(torch.FloatTensor(embedding_matrix))
+VOCAB_SIZE = len(vocab_to_int) + 1  # add 1 for the padding token
+HIDDEN_DIM = 64
+SEQ_LEN = 32
+
+
+
+
 # Load TF-IDF model
 tfidf_model = pickle.load(open('models/modeltfidf.sav', 'rb'))