Update app.py

app.py

@@ -80,7 +80,7 @@ class ImageSearchEngine:
         embedding = F.normalize(embedding, p=2, dim=1)
         return embedding.cpu()
     
-    def search(self, query_text: str, k: int = 10) -> List[Tuple[str, Optional[str]]]:
+    def search(self, query_text: str, k: int = 20) -> List[Tuple[str, Optional[str]]]:
         """Search for images matching the query text."""
         if len(query_text) < 3: # avoid searching for very short queries
             return []
@@ -125,20 +125,14 @@ class GalleryUI:
                 label="Search Query",
                 elem_classes="search-input",
                 autofocus=True,
-                container=False
+                container=False,
+                interactive=True
             )
-            
-            with gr.Row(visible=False): # this is hidden for now, but you can show it if you want
-                self.number_of_results = gr.Dropdown(
-                    choices=[4,8,12,16,24,30],
-                    value=30,
-                    label="Results per page",
-                    elem_classes="dropdown"
-                )
                 
             self.gallery = gr.Gallery(
                 label="Search Results",
-                columns=3,
+                columns=[4],
+                height=600,
                 object_fit="cover",
                 elem_classes="gallery",
                 container=False,
@@ -154,25 +148,22 @@ class GalleryUI:
     
     def _setup_callbacks(self, demo: gr.Blocks) -> None:
         """Setup the interface callbacks."""
-        self.query_text.change(
+        self.query_text.submit(
             self.search_engine.search,
-            inputs=[self.query_text, self.number_of_results],
+            inputs=[self.query_text],#, self.number_of_results],
             outputs=self.gallery,
             show_progress='hidden',
         )
-        
-        self.number_of_results.change(
-            self.search_engine.search,
-            inputs=[self.query_text, self.number_of_results],
-            outputs=self.gallery
-        )
-   
-if __name__ == "__main__":
-    search_engine = ImageSearchEngine(
+
+
+
+search_engine = ImageSearchEngine(
         model_name = "sentence-transformers/all-MiniLM-L6-v2",
         output_dim = 64,
         gallery_folder = "photos",
     )
-    ui = GalleryUI(search_engine)
-    demo = ui.create_interface()
+ui = GalleryUI(search_engine)
+demo = ui.create_interface()
+
+if __name__ == "__main__":
     demo.launch()