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

Update clip_model.py

Browse files
Files changed (1) hide show
  1. clip_model.py +3 -3
clip_model.py CHANGED
@@ -319,13 +319,13 @@ def find_matches(model, image_embeddings, query, image_filenames, n=9):
319
 
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")
327
  return find_matches(model,
328
  image_embeddings,
329
- query=query_text,
330
- image_filenames = valid_df['image'].values,
331
  n=9)
 
319
 
320
 
321
  def clip_image_search(model,image_embeddings,
322
+ query,
323
  image_filenames,
324
  n=9 ):
325
  _, valid_df = make_train_valid_dfs()
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)