File size: 1,438 Bytes
11c750c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import logging
import random
import gradio as gr
class SearchEngineInterface():
def __init__(self, host_ip_address, port_number):
self.logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)
self.host_ip_address = host_ip_address
self.port_number = port_number
def set_search_button_handler_func(self, handler_func):
self.search_button_handler_func = handler_func
def build(self):
self.main_app = gr.Interface(
fn=self.search_button_handler_func,
inputs=[
gr.Textbox(label="Search Query"),
gr.Image(type="pil", label="Upload Image"),
gr.Radio(["Text", "Image"], value="Text", label="Type of prompt", info="Do you want text or image search?"),
gr.Slider(minimum=1, maximum=50, step=1, value=5, label="Number of Results")
],
outputs=gr.Gallery(label="Search Results", columns=3),
title="CLIP Image Search",
description="Enter a text query or upload an image to search for similar images." + " " + \
"The demo uses a smaller version of the COCO dataset (https://cocodataset.org/#home)",
allow_flagging="never"
)
def update(self, search_result):
pass
def run(self):
self.main_app.launch(server_name=self.host_ip_address, server_port=self.port_number) |