Spaces:
Runtime error
Runtime error
Update clip_model.py
Browse files- clip_model.py +1 -1
clip_model.py
CHANGED
@@ -320,7 +320,7 @@ def find_matches(model, image_embeddings, query, image_filenames, n=9):
|
|
320 |
|
321 |
def clip_image_search(model,image_embeddings,
|
322 |
query="dogs on the grass",
|
323 |
-
image_filenames
|
324 |
n=9 ):
|
325 |
_, valid_df = make_train_valid_dfs()
|
326 |
model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
|
|
|
320 |
|
321 |
def clip_image_search(model,image_embeddings,
|
322 |
query="dogs on the grass",
|
323 |
+
image_filenames,
|
324 |
n=9 ):
|
325 |
_, valid_df = make_train_valid_dfs()
|
326 |
model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
|