Spaces:

SeemG
/

TextToImageFlickrSearch

Runtime error

SeemG commited on Jul 27, 2024

Commit

81c03e5

verified ·

1 Parent(s): 4cfe7d1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def generate_images(text, num_images=6):
     # generated_images = clip_image_search(text)
     generated_images = clip_image_search(model,
              image_embeddings,
-             query="desert food",
              image_filenames=valid_df['image'].values,
              n=9)

     # generated_images = clip_image_search(text)
     generated_images = clip_image_search(model,
              image_embeddings,
+             text,
              image_filenames=valid_df['image'].values,
              n=9)