andreped commited on
Commit
963a894
1 Parent(s): f12cbf7

Renamed inference script + reverted resources path in Dockerfile

Browse files
Dockerfile CHANGED
@@ -54,9 +54,9 @@ COPY --chown=user . $HOME/app
54
 
55
  # Download pretrained models
56
  RUN wget "https://github.com/raidionics/Raidionics-models/releases/download/1.2.0/Raidionics-CT_Airways-ONNX-v12.zip" && \
57
- unzip "Raidionics-CT_Airways-ONNX-v12.zip" && mkdir -p demo/resources/models/ && mv CT_Airways/ demo/resources/models/CT_Airways/
58
  RUN wget "https://github.com/raidionics/Raidionics-models/releases/download/1.2.0/Raidionics-CT_Lungs-ONNX-v12.zip" && \
59
- unzip "Raidionics-CT_Lungs-ONNX-v12.zip" && mv CT_Lungs/ demo/resources/models/CT_Lungs/
60
 
61
  RUN rm -r *.zip
62
 
 
54
 
55
  # Download pretrained models
56
  RUN wget "https://github.com/raidionics/Raidionics-models/releases/download/1.2.0/Raidionics-CT_Airways-ONNX-v12.zip" && \
57
+ unzip "Raidionics-CT_Airways-ONNX-v12.zip" && mkdir -p resources/models/ && mv CT_Airways/ resources/models/CT_Airways/
58
  RUN wget "https://github.com/raidionics/Raidionics-models/releases/download/1.2.0/Raidionics-CT_Lungs-ONNX-v12.zip" && \
59
+ unzip "Raidionics-CT_Lungs-ONNX-v12.zip" && mv CT_Lungs/ resources/models/CT_Lungs/
60
 
61
  RUN rm -r *.zip
62
 
demo/src/gui.py CHANGED
@@ -2,7 +2,7 @@ import os
2
 
3
  import gradio as gr
4
 
5
- from .compute import run_model
6
  from .utils import load_ct_to_numpy
7
  from .utils import load_pred_volume_to_numpy
8
  from .utils import nifti_to_glb
 
2
 
3
  import gradio as gr
4
 
5
+ from .inference import run_model
6
  from .utils import load_ct_to_numpy
7
  from .utils import load_pred_volume_to_numpy
8
  from .utils import nifti_to_glb
demo/src/{compute.py → inference.py} RENAMED
File without changes