SeemG commited on
Commit
7fe0bf3
·
verified ·
1 Parent(s): b837ea5

Update clip_model.py

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