Update pages/imdb.py

pages/imdb.py

@@ -14,6 +14,13 @@ 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))
@@ -28,11 +35,6 @@ for word, i in vocab_to_int.items():
 # Load TF-IDF model
 tfidf_model = pickle.load(open('models/modeltfidf.sav', 'rb'))
 
-# 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
 
 
 class LSTMClassifierBi32(nn.Module):