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