Upload 12 files

.gitattributes

@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+sa_1309.jpg filter=lfs diff=lfs merge=lfs -text
+sa_192.jpg filter=lfs diff=lfs merge=lfs -text
+sa_414.jpg filter=lfs diff=lfs merge=lfs -text
+sa_862.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -17,11 +17,19 @@ from PIL import Image as PILIMAGE
 
 from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
 from sentence_transformers import SentenceTransformer, util
-
-
-
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+examples = [["fruits.jpg"],
+            ["flowers.jpg"],
+            ["sa_10039.jpg"],
+            ["dogs.jpg"],
+            ["sa_8776.jpg"],
+            ["sa_561.jpg"],
+            ["sa_11025.jpg"],
+            ["sa_1309.jpg"],
+            ["sa_192.jpg"],
+            ["sa_862.jpg"],["sa_414.jpg"]]
+
 # Define model
 model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32").to(device)
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
@@ -104,7 +112,7 @@ gr.Interface(fn=find_best_matches,
             gr.Radio([T2I, I2I]),
             gr.Textbox(lines=1, label="Text query", placeholder="Introduce the search text...",
             )],
-            theme="grass",
+            theme="grass",examples=examples,
             outputs=[gr.Gallery(label="Generated images", show_label=False, elem_id="gallery", scale=2)],title="CLIP Image Search",
         description="This application displays TOP THREE images from Unsplash dataset that best match the search query provided by the user. Moreover, the input can be provided via two modes ie text or image form.").launch()