Renamed inference script + reverted resources path in Dockerfile
Browse files- Dockerfile +2 -2
- demo/src/gui.py +1 -1
- demo/src/{compute.py → inference.py} +0 -0
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
|
| 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/
|
| 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 .
|
| 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
|