import gradio as gr import os from PIL import Image from inference import Inference inference = Inference() with gr.Blocks() as block: options = gr.Dropdown(choices=["Blip Saffal", "Blip CN"], label="Models", info="Select the model to use..", ) # need to improve this one... txt = gr.Textbox(label="Insert a question...", lines=2) txt_3 = gr.Textbox(value="", label="Your answer is here..") btn = gr.Button(value="Submit") frame_path = os.path.join(os.path.dirname(__file__), "testing.jpg") frame = Image.open(frame_path) inference.logger.info(f"Image loaded {frame.size}") image = gr.Image(type="pil", value=frame) btn.click(inference.inference, inputs=[options, image, txt], outputs=[txt_3]) if __name__ == "__main__": block.launch()