Update app.py

app.py

@@ -67,32 +67,11 @@ 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"
 
-    # Use a more robust unpickling method
-    try:
-        with open(word_index_temp, "rb") as f:
-            # Try different encoding methods
-            try:
-                word_index_dict = pickle.load(f)
-            except:
-                f.seek(0)  # Reset file pointer
-                word_index_dict = pickle.load(f, encoding='bytes')
-
-    except Exception as e:
-        print(f"Error loading pickle file: {e}")
-        word_index_dict = {}  # Provide a fallback empty dictionary
+    # Load word index dictionary
+    word_index_dict = pickle.load(open(word_index_temp, "rb"), encoding="latin")
 
     # Load embeddings numpy
     embeddings = np.load(embeddings_temp)