bala1802 commited on
Commit
8b3f764
1 Parent(s): d4e8957

Update clip_inferencing.py

Browse files
Files changed (1) hide show
  1. clip_inferencing.py +2 -2
clip_inferencing.py CHANGED
@@ -55,11 +55,11 @@ def find_matches(model, image_embeddings, query, image_filenames, n=9):
55
 
56
  plt.show()
57
 
58
- def inference():
59
  valid_df = load_df()
60
  image_embeddings = load_image_embeddings()
61
  find_matches(load_model(model_path="model/best.pt"),
62
  image_embeddings,
63
- query="dogs on the grass",
64
  image_filenames=valid_df['image'].values, n=9)
65
 
 
55
 
56
  plt.show()
57
 
58
+ def inference(query):
59
  valid_df = load_df()
60
  image_embeddings = load_image_embeddings()
61
  find_matches(load_model(model_path="model/best.pt"),
62
  image_embeddings,
63
+ query=query,
64
  image_filenames=valid_df['image'].values, n=9)
65