SeemG commited on
Commit
4cfe7d1
·
verified ·
1 Parent(s): 4a5d321

Update clip_model.py

Browse files
Files changed (1) hide show
  1. clip_model.py +1 -1
clip_model.py CHANGED
@@ -326,6 +326,6 @@ def clip_image_search(model,image_embeddings,
326
  model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
327
  return find_matches(model,
328
  image_embeddings,
329
- query="dogs on the grass",
330
  image_filenames = valid_df['image'].values,
331
  n=9)
 
326
  model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
327
  return find_matches(model,
328
  image_embeddings,
329
+ query=query,
330
  image_filenames = valid_df['image'].values,
331
  n=9)