Update search_ocean.py

search_ocean.py

@@ -8,6 +8,13 @@ import torch
 text_index = faiss.read_index("text_recipes_test_br_00.ocean")
 image_index = faiss.read_index("image_recipes_test_br_00.ocean")
 
+# Load metadata
+with open("text_metadata.json", "r", encoding="utf-8") as f:
+    text_metadata = json.load(f)
+
+with open("image_metadata.json", "r", encoding="utf-8") as f:
+    image_metadata = json.load(f)
+
 # Initialize models for text and image processing
 text_encoder = SentenceTransformer("sentence-transformers/all-MiniLM-L12-v2")
 clip_model, clip_preprocess = clip.load("ViT-B/32", device="cpu")