Update app.py

app.py

@@ -14,6 +14,17 @@ from huggingface_hub import from_pretrained_keras
 
 model = from_pretrained_keras("keras-io/text-generation-miniature-gpt")
 
+a = []
+word_to_index = {}
+
+a_file = open("aclImdb/imdb.vocab")
+for line in a_file:
+  a.append(line.strip())
+print(len(a))
+
+for index, word in enumerate(a):
+    word_to_index[index] = word
+
 def text_process_pipeline(text): #pipeline
   tokenizer = Tokenizer(num_words=80, split=' ')
   tokenizer.fit_on_texts(word_to_index.values())