Update GPT4o_class.py

GPT4o_class.py

@@ -45,12 +45,12 @@ class GPT4o:
         self.model.to(self.device)
 
         # Load the embeddings and coordinates from the pickle file
-        with open('StreetCLIP_1m_merged.pkl', 'rb') as f: # Enter the path to the pickle file
-            self.MP_16_Embeddings = pickle.load(f)
-            self.locations = [value['location'] for key, value in self.MP_16_Embeddings.items()]
+        with open('StreetCLIP_5m_merged.pkl', 'rb') as f: # Enter the path to the pickle file
+            self.Embeddings = pickle.load(f)
+            self.locations = [value['location'] for key, value in self.Embeddings.items()]
 
         # Load the Faiss index
-        index2 = faiss.read_index("StreetCLIP_1m_merged.bin") # Enter the path to the Faiss index file
+        index2 = faiss.read_index("StreetCLIP_5m_merged.bin") # Enter the path to the Faiss index file
         self.gpu_index = index2
 
     def read_image(self, image_path):