Spaces:

clip-italian
/

clip-italian-demo

Sleeping

Silvia Terragni commited on Jul 18, 2021

Commit

f920e44

1 Parent(s): 6985586

fix find_image

Files changed (1) hide show

text2image.py CHANGED Viewed

@@ -99,7 +99,7 @@ def app():
             dataset = utils.CustomDataSet("photos/", transform=val_preprocess)
-            image_paths = utils.find_imageread_markdown_file(
                 query, model, dataset, tokenizer, image_features, n=2
             )

             dataset = utils.CustomDataSet("photos/", transform=val_preprocess)
+            image_paths = utils.find_image(
                 query, model, dataset, tokenizer, image_features, n=2
             )