model examples and ui improvement

app.py

@@ -63,10 +63,12 @@ def inference(input):
 
 examples = [
     ['examples/BRATS_225.nii.gz', 83, 2],
+    ['examples/BRATS_485.nii.gz', 90, 1],
+    ['examples/BRATS_485.nii.gz', 110, 0]
 ]
 
 
-def sepia(tumor_file, slice_number, channel, audio_question):
+def detector(tumor_file, slice_number, channel, audio_question):
     tumor_file_path = tumor_file.name
     processed_data = preproc_transforms({'image': [tumor_file_path]})
     tensor_3d_input = processed_data['image'].unsqueeze(0).to('cpu')
@@ -88,13 +90,19 @@ def sepia(tumor_file, slice_number, channel, audio_question):
     return (channel_image, segment_image, "Question answer")
 
 
-interface = gr.Interface(fn=sepia, inputs=[gr.File(label="Tumor File"),
-                                           gr.Slider(0, 200, 50, step=1, label="Slice Number"),
-                                           gr.Slider(0, 2, 1, step=1, label="Channel Number"),
-                                           gr.Audio(source="microphone"), ],
+interface = gr.Interface(fn=detector, inputs=[gr.File(label="Tumor File"),
+                                              gr.Slider(0, 200, 50, step=1, label="Slice Number"),
+                                              gr.Radio([0, 1, 2], label="Channel"),
+                                              gr.Audio(source="microphone"), ],
                          outputs=[gr.Image(label='channel', shape=(1, 1)),
                                   gr.Image(label='Segmented Tumor', shape=(1, 1)),
                                   gr.Textbox(label="Medical Summary")], title=title,
+                         examples=examples,
                          description=description, outputs_layout="row", theme='dark')
 
+theme = gr.themes.Default().set(
+    button_primary_background_fill="#FF0000",
+    button_primary_background_fill_dark="#AAAAAA",
+)
+
 interface.launch(debug=True)

examples/BRATS_109.nii.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4873bc09872391983979b10cfccd3c5bb30b84ff2c824cb4130cb7a7dfb1f1ec
+size 8590962

examples/BRATS_225.nii.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e2757295889234461bb24cbe499f1eafdc6b2555147308e7f2c62025db5531d8
+size 10837716

examples/BRATS_485.nii.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1de6be1eeb49c788baa286a21d71546b2974bc300d5bc6ce4541e41854a0fefb
+size 8327084