mekosotto Claude Opus 4.7 (1M context) commited on
Commit
e82971e
·
1 Parent(s): 8ec95a0

deps: add torch/torchvision/pillow for MRI DL 2D

Browse files

Pinned torch==2.4.1, torchvision==0.19.1, pillow==10.4.0 to enable the
2D resnet18 4-class Alzheimer's MRI inference path. These pull no
duplicate transitive deps that conflict with onnxruntime; both backends
coexist for MRI prediction (selected via MRI_MODEL_KIND env in upcoming
commits).

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Files changed (1) hide show
  1. requirements.txt +3 -0
requirements.txt CHANGED
@@ -32,6 +32,9 @@ mlflow==2.16.0
32
  shap==0.46.0
33
  joblib==1.4.2
34
  onnxruntime==1.19.2 # MRI volumetric ONNX inference (external DL artifact)
 
 
 
35
 
36
  # --- Tooling / tests ---
37
  pytest==8.3.3
 
32
  shap==0.46.0
33
  joblib==1.4.2
34
  onnxruntime==1.19.2 # MRI volumetric ONNX inference (external DL artifact)
35
+ torch==2.4.1 # MRI 2D resnet18 inference (Plan: external-assets-integration)
36
+ torchvision==0.19.1 # resnet18 + image transforms
37
+ pillow==10.4.0 # image loading for 2D MRI inputs
38
 
39
  # --- Tooling / tests ---
40
  pytest==8.3.3