Spaces:
Runtime error
Runtime error
Update clip_model.py
Browse files- 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
|
| 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=
|
| 330 |
-
image_filenames =
|
| 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)
|