Update app.py

app.py

@@ -80,9 +80,19 @@ 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
-    with open(word_index_temp, "rb") as f:
-        word_index_dict = pickle.load(f, encoding="latin1")
+    # 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 embeddings numpy
     embeddings = np.load(embeddings_temp)