Update app.py

app.py

@@ -37,19 +37,22 @@ def persistent_load(pers_id):
         raise HTTPException(status_code=500, detail=f"Unknown persistent ID: {pers_id}")
 
 
-# Function to load models
 def load_models():
     try:
-        # Load embeddings data with custom persistent_load function
-        with open("embeddings.pkl", "rb") as file:
-                unpickler = pickle.Unpickler(file)
-                unpickler.persistent_load = persistent_load
-                global_models.embeddings_data = unpickler.load()
-                print("Embeddings data loaded successfully.")
-    except pickle.UnpicklingError as e:
-        raise HTTPException(status_code=500, detail=f"Unpickling error: {e}")
+        with open('embeddings.pkl', 'rb') as f:
+            embeddings_data = pickle.load(f, encoding='latin1')  # or 'bytes'
+        
+        # If embeddings_data is a dictionary, check its content
+        if isinstance(embeddings_data, dict):
+            print("Loaded embeddings dictionary")
+        
+        # Proceed with your logic using embeddings_data
+        # For example, assign to global models or something similar
+        global_models.embeddings_data = embeddings_data
+    
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to load models: {e}")
+        print(f"Error loading embeddings data: {e}")
+        raise HTTPException(status_code=500, detail="Failed to load embeddings data.")
         
 app = FastAPI()