Jyothirmai commited on
Commit
f85b191
·
verified ·
1 Parent(s): 436aa77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -18,18 +18,18 @@ def generate_caption_clipgpt(image):
18
 
19
  # Sample image paths
20
  sample_images = [
21
- "image-captioning-chest-xrays/CXR191_IM-0591-1001.jpg",
22
- "image-captioning-chest-xrays/CXR191_IM-0598-1001.jpg",
23
- "image-captioning-chest-xrays/CXR191_IM-0601-1001.jpg",
24
- "image-captioning-chest-xrays/CXR191_IM-0609-1001.jpg",
25
- "image-captioning-chest-xrays/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(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():