Update clip_inferencing.py
Browse files- 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=
|
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 |
|