| import gradio as gr | |
| from PIL import Image # For image handling | |
| # Replace with paths or loading functions for your specific models | |
| def load_model_1(): | |
| # ... load your first model | |
| return model_1 | |
| def load_model_2(): | |
| # ... load your second model | |
| return model_2 | |
| def load_model_3(): | |
| # ... load your third model | |
| return model_3 | |
| def generate_caption(model, image): | |
| # ... perform inference with your model | |
| return caption | |
| # models = [load_model_1(), load_model_2(), load_model_3()] | |
| with gr.Blocks() as demo: | |
| with gr.Row(): | |
| image = gr.Image(label="Upload Chest X-ray") | |
| with gr.Row(): | |
| gr.Radio(["Model 1", "Model 2", "Model 3"], label="Select Model") | |
| with gr.Row(): | |
| caption = gr.Textbox(label="Generated Caption") | |
| # image.change( | |
| # fn=generate_caption, | |
| # inputs=[image, gr.inputs.Radio], | |
| # outputs=caption | |
| # ) | |
| demo.launch() | |