add app.py

app.py

@@ -106,13 +106,28 @@ with gr.Blocks(title="Spleen 3D segmentation with MONAI - ClassCat",
             ex_image3 = gr.Image(value='thumbnails/val_image3.png', interactive=False, label='ex 4')
             ex_image3.style(width=128, height=128)
 
+       with gr.Column():
+            ex_btn4 = gr.Button("Example 5")
+            ex_btn4.style(full_width=False, css="width:20px;")
+            ex_image4 = gr.Image(value='thumbnails/val_image4.png', interactive=False, label='ex 5')
+            ex_image4.style(width=128, height=128)
+
+       with gr.Column():
+            ex_btn5 = gr.Button("Example 6")
+            ex_btn5.style(full_width=False, css="width:20px;")
+            ex_image5 = gr.Image(value='thumbnails/val_image5.png', interactive=False, label='ex 6')
+            ex_image5.style(width=128, height=128)
+
     ex_btn0.click(fn=load_image0, outputs=[sample_index, input_image, label_image])
     ex_btn1.click(fn=load_image1, outputs=[sample_index, input_image, label_image])
     ex_btn2.click(fn=load_image2, outputs=[sample_index, input_image, label_image])
     ex_btn3.click(fn=load_image3, outputs=[sample_index, input_image, label_image])
+    ex_btn4.click(fn=load_image4, outputs=[sample_index, input_image, label_image])
+    ex_btn5.click(fn=load_image5, outputs=[sample_index, input_image, label_image])
+
 
     gr.HTML("""<br/>""")
-    gr.HTML("""<h4 style="color:navy;">2. Then, click "Infer" button to predict segmentation images. It will take about 30 seconds (on cpu)</h4>""")
+    gr.HTML("""<h4 style="color:navy;">2. Then, click "Infer" button to predict a segmentation image. It will take about 15 seconds (on cpu)</h4>""")
 
     send_btn = gr.Button("Infer")
     send_btn.click(fn=predict, inputs=[sample_index], outputs=[output_image])