Update app.py
Browse files
app.py
CHANGED
@@ -18,18 +18,18 @@ def generate_caption_clipgpt(image):
|
|
18 |
|
19 |
# Sample image paths
|
20 |
sample_images = [
|
21 |
-
"
|
22 |
-
"
|
23 |
-
"
|
24 |
-
"
|
25 |
-
"
|
26 |
]
|
27 |
|
28 |
# Gradio interface
|
29 |
with gr.Blocks() as demo:
|
30 |
with gr.Row():
|
31 |
image = gr.Image(label="Upload Chest X-ray")
|
32 |
-
sample_images_gallery = gr.Gallery(sample_images, label="Sample Images")
|
33 |
with gr.Row():
|
34 |
model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
|
35 |
with gr.Row():
|
|
|
18 |
|
19 |
# Sample image paths
|
20 |
sample_images = [
|
21 |
+
"CXR191_IM-0591-1001.jpg",
|
22 |
+
"CXR191_IM-0598-1001.jpg",
|
23 |
+
"CXR191_IM-0601-1001.jpg",
|
24 |
+
"CXR191_IM-0609-1001.jpg",
|
25 |
+
"CXR191_IM-0618-1001.jpg"
|
26 |
]
|
27 |
|
28 |
# Gradio interface
|
29 |
with gr.Blocks() as demo:
|
30 |
with gr.Row():
|
31 |
image = gr.Image(label="Upload Chest X-ray")
|
32 |
+
sample_images_gallery = gr.Gallery(value=sample_images, label="Sample Images", columns=5)
|
33 |
with gr.Row():
|
34 |
model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
|
35 |
with gr.Row():
|