Full path to model in docker img
Browse files- demo/app.py +1 -1
demo/app.py
CHANGED
|
@@ -29,7 +29,7 @@ def nifti_to_glb(path):
|
|
| 29 |
def run_model(input_path):
|
| 30 |
from livermask.utils.run import run_analysis
|
| 31 |
|
| 32 |
-
run_analysis(cpu=True, extension='.nii', path=input_path, output='prediction', verbose=True, vessels=False, name="
|
| 33 |
|
| 34 |
#cmd_docker = ["python3", "-m", "livermask.livermask", "--input", input_path, "--output", "prediction", "--verbose"]
|
| 35 |
#sp.check_call(cmd_docker, shell=True) # @FIXME: shell=True here is not optimal -> starts a shell after calling script
|
|
|
|
| 29 |
def run_model(input_path):
|
| 30 |
from livermask.utils.run import run_analysis
|
| 31 |
|
| 32 |
+
run_analysis(cpu=True, extension='.nii', path=input_path, output='prediction', verbose=True, vessels=False, name="/home/user/app/model.h5", mp_enabled=False)
|
| 33 |
|
| 34 |
#cmd_docker = ["python3", "-m", "livermask.livermask", "--input", input_path, "--output", "prediction", "--verbose"]
|
| 35 |
#sp.check_call(cmd_docker, shell=True) # @FIXME: shell=True here is not optimal -> starts a shell after calling script
|