Update utils.py

utils.py

@@ -11,6 +11,8 @@ from transformers import (
     BlipProcessor,
     BlipForConditionalGeneration
 )
+import zipfile
+import os
 
 MEAN = [0.48145466, 0.4578275, 0.40821073]
 STD = [0.26862954, 0.26130258, 0.27577711]
@@ -22,9 +24,15 @@ def load_models(device):
     blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
     return clip_model, clip_processor, blip_model, blip_processor
 
-def load_data(parquet_path="src/food101_embeddings_10000.parquet"):
-    table = pq.read_table(parquet_path)
-    df = table.to_pandas()
+
+def load_data(parquet_path="food101_embeddings_10000.parquet"):
+    
+    if not os.path.exists("food_images"):
+        with zipfile.ZipFile("food_images_10000.zip", "r") as zip_ref:
+            zip_ref.extractall("food_images")
+
+    df = pd.read_parquet(parquet_path)
+    df["image_path"] = df["image_path"].apply(lambda p: os.path.join("food_images", os.path.basename(p)))
     embeddings = np.vstack(df["embedding"].to_numpy())
     return df, embeddings