Update app.py

app.py

@@ -67,11 +67,22 @@ def download_glove_embeddings_gdrive(model_type):
 
 # @st.cache_data()
 def load_glove_embeddings_gdrive(model_type):
+    # word_index_temp = "word_index_dict_" + str(model_type) + "_temp.pkl"
+    # embeddings_temp = "embeddings_" + str(model_type) + "_temp.npy"
+
+    # # Load word index dictionary
+    # word_index_dict = pickle.load(open(word_index_temp, "rb"), encoding="latin")
+
+    # # Load embeddings numpy
+    # embeddings = np.load(embeddings_temp)
+
+    # return word_index_dict, embeddings
     word_index_temp = "word_index_dict_" + str(model_type) + "_temp.pkl"
     embeddings_temp = "embeddings_" + str(model_type) + "_temp.npy"
 
     # Load word index dictionary
-    word_index_dict = pickle.load(open(word_index_temp, "rb"), encoding="latin")
+    with open(word_index_temp, "rb") as f:
+        word_index_dict = pickle.load(f, encoding="latin1")
 
     # Load embeddings numpy
     embeddings = np.load(embeddings_temp)