|
import gradio as gr |
|
from PIL import Image |
|
|
|
|
|
def load_model_1(): |
|
|
|
return model_1 |
|
|
|
def load_model_2(): |
|
|
|
return model_2 |
|
|
|
def load_model_3(): |
|
|
|
return model_3 |
|
|
|
def generate_caption(model, image): |
|
|
|
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() |
|
|