Niko.Koutsoubis commited on
Commit
a3386b3
·
1 Parent(s): f369cf3

Update requirements.txt for embedding extraction pipeline

Browse files
Files changed (1) hide show
  1. requirements.txt +27 -7
requirements.txt CHANGED
@@ -1,9 +1,29 @@
 
 
 
 
 
 
 
1
  torch>=2.0.0
2
  torchvision>=0.15.0
3
- transformers>=4.30.0
4
- pydicom>=2.3.0
5
- torchio>=0.19.0
6
- SimpleITK>=2.2.0
7
- numpy>=1.21.0
8
- Pillow>=9.0.0
9
- sybil>=1.2.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core dependencies for Sybil embedding extraction
2
+ # Install with: pip install -r requirements.txt
3
+
4
+ # HuggingFace Hub for model download
5
+ huggingface_hub>=0.20.0
6
+
7
+ # PyTorch for model inference
8
  torch>=2.0.0
9
  torchvision>=0.15.0
10
+
11
+ # Data processing
12
+ numpy>=1.24.0
13
+ pandas>=2.0.0
14
+
15
+ # DICOM file handling
16
+ pydicom>=2.4.0
17
+
18
+ # Image processing (required by Sybil model)
19
+ Pillow>=10.0.0
20
+
21
+ # Model utilities
22
+ scikit-learn>=1.3.0
23
+ tqdm>=4.65.0
24
+
25
+ # Optional but recommended for performance
26
+ pyarrow>=14.0.0
27
+
28
+ # Note: For GPU support with CUDA 11.8, install PyTorch with:
29
+ # pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118