Update app.py

app.py

@@ -39,17 +39,13 @@ def get_clip_embeddings(input_data, input_type='text'):
 
     return embeddings.numpy()
 
-
 veggies = load_dataset('vojtam/vegetables')
 with open('img_embeddings.pkl', 'rb') as file:
     img_embeddings = pickle.load(file)
 
 cos = nn.CosineSimilarity(dim=1, eps=1e-6)
 
-text = gr.Textbox(label = "Enter the description of the images you want to search for", placeholder='Your text goes here')
-image = gr.Gallery()
-
-def get_similar_images(text, n = 4):
+def get_similar_images(text, n=4):
     if text:
         text_embedding = get_clip_embeddings(text, input_type='text')
         sims = cos(torch.tensor(text_embedding), torch.tensor(img_embeddings))
@@ -59,8 +55,32 @@ def get_similar_images(text, n = 4):
         for index in top_n:
             imgs.append(veggies['train'][index.item()]['image'])
         return imgs
+    return []
+
+css = """
+.full-height-gallery {
+    height: calc(100vh - 250px);
+    overflow-y: auto;
+}
+
+#submit-btn {
+    background-color: #ff5b00;
+    color: #ffffff;
+}
+"""
+
+with gr.Blocks(css=css) as intf:
+    with gr.Row():
+        text_input = gr.Textbox(label="Enter the description of the images you want to search for", placeholder='Your text goes here')
+    with gr.Row():
+        submit_btn = gr.Button("Submit", elem_id="submit-btn")
+        clear_btn = gr.Button("Clear")
+    with gr.Row():
+        gallery = gr.Gallery(label="Similar Images", show_label=False, elem_classes = ["full-height-gallery"])
+    
+    submit_btn.click(fn=get_similar_images, inputs=text_input, outputs=gallery)
+    clear_btn.click(fn=lambda: [None, []], inputs=None, outputs=[text_input, gallery])
     
 
 
-intf = gr.Interface(fn = get_similar_images, inputs = text, outputs = image)
 intf.launch(share=True)