Update app.py

README.md

@@ -1,5 +1,5 @@
 ---
-title: Glioma Segmentation
+title: Brain Tumor Segmentation
 emoji: 🧠
 colorFrom: indigo
 colorTo: red

app.py

@@ -16,7 +16,9 @@ from monai.transforms import (
 BUNDLE_NAME = 'spleen_ct_segmentation_v0.1.0'
 BUNDLE_PATH = os.path.join(torch.hub.get_dir(), 'bundle', BUNDLE_NAME)
 
-#examples = ['examples/BRATS_485.nii.gz']
+examples = [
+    ['examples/BRATS_485.nii.gz'],
+]
 
 model, _, _ = bundle.load(
     name = BUNDLE_NAME,
@@ -69,7 +71,7 @@ def predict(input_file, z_axis, model=model, device=device):
     #pred_2_image = pred_image[0, 1, :, :, z_axis]
     #pred_3_image = pred_image[0, 2, :, :, z_axis]
     
-    return input_t1c_image, pred_1_image
+    return input_t1c_image, pred_1_image, z_axis
 
 iface = gr.Interface(
     fn=predict,
@@ -79,9 +81,10 @@ iface = gr.Interface(
     ], 
     outputs=[
         gr.Image(label='input image'),
-        gr.Image(label='segmentation')],
-    title='Segment Gliomas using MONAI',
-    #examples=examples,
+        gr.Image(label='segmentation'),
+        gr.Slider(0, 200, label='z-axis', value=100)],
+    title='Segment Brain Tumors using MONAI',
+    examples=examples,
 )
 
 iface.launch()