Update utils.py

utils.py

@@ -63,7 +63,7 @@ def run(tmpdirname):
     if st.session_state.option is not None:
         image = Path(__file__).parent / str(st.session_state.option)
 
-        inference.infer([image], tmpdirname, [0], split_level=1)
+        inference.infer([image], tmpdirname, [0], split_level=2, cpu_only=True)
         seg_name = add_postfix(image.name, "seg")
         preds_path = tmpdirname + "/" + seg_name
         st.session_state.preds_3D = read_image(preds_path)