Decrease workload on CPUs

app.py

@@ -72,9 +72,10 @@ def save_file(segmentation, path):
 def infer(image_path):
     with tempfile.TemporaryDirectory() as tmpdirname:
 
-        inference.infer(
-            [image_path], tmpdirname, [0, 1, 2, 3, 4], cpu_only=False if torch.cuda.is_available() else True
-        )
+        if torch.cuda.is_available():
+            inference.infer([image_path], tmpdirname, [0, 1, 2, 3, 4], cpu_only=False, split_level=1)
+        else:
+            inference.infer([image_path], tmpdirname, [0], cpu_only=True, split_level=1)
         filename = add_postfix(Path(image_path).name, "seg")
         segmentation = sitk.ReadImage(tmpdirname + "/" + filename)
 
@@ -124,13 +125,6 @@ with gr.Blocks(css=css, title="MRSegmentator") as iface:
         with gr.Column():
             overlay_image_np = gr.AnnotatedImage(label="Axial View")
 
-            # pred_dict = gr.Label(label="Prediction")
-            # explanation= gr.Textbox(label="Classification Decision")
-
-            # with gr.Accordion("Additional Information", open=False):
-            #     gradcam = gr.Image(label="GradCAM")
-            #     cropped_boxed_array_disp = gr.Image(label="Bounding Box")
-
     input_file.change(
         utils.read_and_display,
         inputs=[input_file, image_state, seg_state],
@@ -149,5 +143,4 @@ with gr.Blocks(css=css, title="MRSegmentator") as iface:
 
 if __name__ == "__main__":
     iface.queue()
-    # iface.launch(server_name='0.0.0.0', server_port=8080)
     iface.launch()