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

Update clip_model.py

Browse files
Files changed (1) hide show
  1. clip_model.py +3 -3
clip_model.py CHANGED
@@ -322,10 +322,10 @@ 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")
327
  return find_matches(model,
328
  image_embeddings,
329
  query=query_text,
330
- image_filenames=valid_df['image'].values,
331
  n=9)
 
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")
327
  return find_matches(model,
328
  image_embeddings,
329
  query=query_text,
330
+ image_filenames = valid_df['image'].values,
331
  n=9)