Silvia Terragni commited on
Commit
f920e44
1 Parent(s): 6985586

fix find_image

Browse files
Files changed (1) hide show
  1. text2image.py +1 -1
text2image.py CHANGED
@@ -99,7 +99,7 @@ def app():
99
 
100
  dataset = utils.CustomDataSet("photos/", transform=val_preprocess)
101
 
102
- image_paths = utils.find_imageread_markdown_file(
103
  query, model, dataset, tokenizer, image_features, n=2
104
  )
105
 
 
99
 
100
  dataset = utils.CustomDataSet("photos/", transform=val_preprocess)
101
 
102
+ image_paths = utils.find_image(
103
  query, model, dataset, tokenizer, image_features, n=2
104
  )
105