Update app.py

app.py

@@ -109,7 +109,8 @@ def update_kind(selected_key, selected_organ):
     return gr.update(choices=kinds, value=kinds)
 
 
-def generate_image(organ, kind, keys):
+def generate_image(organ, kinds, keys):
+    kind = ",".join(kinds)
     image, label = pipeline.generate(organ=organ, kind=kind, keys=keys)
     plt.subplot(1, 2, 1)
     plt.imshow(image)
@@ -132,6 +133,18 @@ with gr.Blocks() as demo:
     generate_button = gr.Button("Generate Image")
     output_image = gr.Image(label="Visualization")
 
+    gr.Examples(
+        examples=[
+            ["CVC-ClinicDB", "polyp colonoscopy", ["polyp"]],
+            ["BUSI", "breast ultrasound", ["breast tumor"]],
+            ["LiTS2017", "abdomen CT scans", ["liver", "liver tumor"]],
+            ["KiTS2019", "abdomen CT scans", ["kidney", "kidney tumor"]],
+            ["ACDC", "cardiovascular ventricle mri", ["right ventricle", "left ventricle","left ventricle"]],
+            ["AMOS2022", "abdomen CT scans", ["liver", "right kidney", "spleen"]]
+        ],
+        inputs=[organ_dropdown, kind_checkbox, keys_dropdown]
+    )
+
     generate_button.click(generate_image, inputs=[organ_dropdown, kind_checkbox, keys_dropdown], outputs=output_image)
 
 demo.launch()