andreped commited on
Commit
6209ec6
1 Parent(s): 55b64e4

Attempt to copy model inside /code container

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. livermask/utils/run.py +1 -0
Dockerfile CHANGED
@@ -34,7 +34,7 @@ RUN apt install wget -y
34
 
35
  # Download pretrained parenchyma model
36
  RUN wget "https://github.com/andreped/livermask/releases/download/trained-models-v1/model.h5"
37
- #COPY --chown=user ./model.h5 /code/model.h5
38
 
39
  # Set up a new user named "user" with user ID 1000
40
  RUN useradd -m -u 1000 user
 
34
 
35
  # Download pretrained parenchyma model
36
  RUN wget "https://github.com/andreped/livermask/releases/download/trained-models-v1/model.h5"
37
+ COPY --chown=user ./model.h5 /code/model.h5
38
 
39
  # Set up a new user named "user" with user ID 1000
40
  RUN useradd -m -u 1000 user
livermask/utils/run.py CHANGED
@@ -70,6 +70,7 @@ def run_analysis(path, output, cpu, verbose, vessels, extension, name=None, name
70
  if multiple_flag:
71
  os.makedirs(output + "/", exist_ok=True)
72
 
 
73
  for curr in tqdm(paths, "CT:"):
74
  # check if current file is a nifti file, if not, skip
75
  if curr.endswith(".nii") or curr.endswith(".nii.gz"):
 
70
  if multiple_flag:
71
  os.makedirs(output + "/", exist_ok=True)
72
 
73
+ log.info("Starting inference...")
74
  for curr in tqdm(paths, "CT:"):
75
  # check if current file is a nifti file, if not, skip
76
  if curr.endswith(".nii") or curr.endswith(".nii.gz"):