Ziyuan111 commited on
Commit
3bbc97b
·
verified ·
1 Parent(s): e57faac

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -22,7 +22,7 @@ def load_models(device):
22
  blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
23
  return clip_model, clip_processor, blip_model, blip_processor
24
 
25
- def load_data(parquet_path="src/food101_embeddings_with_path.parquet"):
26
  table = pq.read_table(parquet_path)
27
  df = table.to_pandas()
28
  embeddings = np.vstack(df["embedding"].to_numpy())
 
22
  blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
23
  return clip_model, clip_processor, blip_model, blip_processor
24
 
25
+ def load_data(parquet_path="src/food101_embeddings_10000.parquet"):
26
  table = pq.read_table(parquet_path)
27
  df = table.to_pandas()
28
  embeddings = np.vstack(df["embedding"].to_numpy())