Spaces:
Runtime error
Runtime error
Update utils.py
Browse files
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/
|
| 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())
|