SeemG commited on
Commit
e1ac9c3
·
verified ·
1 Parent(s): 52d358c

Update clip_model.py

Browse files
Files changed (1) hide show
  1. clip_model.py +1 -1
clip_model.py CHANGED
@@ -320,7 +320,7 @@ def find_matches(model, image_embeddings, query, image_filenames, n=9):
320
 
321
  def clip_image_search(model,image_embeddings,
322
  query="dogs on the grass",
323
- image_filenames=valid_df['image'].values,
324
  n=9 ):
325
  _, valid_df = make_train_valid_dfs()
326
  model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
 
320
 
321
  def clip_image_search(model,image_embeddings,
322
  query="dogs on the grass",
323
+ image_filenames,
324
  n=9 ):
325
  _, valid_df = make_train_valid_dfs()
326
  model, image_embeddings = get_image_embeddings(valid_df, "best.pt")