31puneet commited on
Commit
db828f6
·
1 Parent(s): 19519bf

Fix numpy, torch version and opencv-headless for HF Spaces

Browse files
Files changed (2) hide show
  1. Dockerfile +3 -1
  2. requirements.txt +3 -5
Dockerfile CHANGED
@@ -10,7 +10,9 @@ RUN pip install --no-cache-dir --upgrade pip
10
 
11
  COPY --chown=user requirements.txt requirements.txt
12
 
13
- RUN pip install --no-cache-dir torch==2.1.0+cpu torchvision==0.16.0+cpu --index-url https://download.pytorch.org/whl/cpu
 
 
14
 
15
  RUN pip install --no-cache-dir -r requirements.txt
16
 
 
10
 
11
  COPY --chown=user requirements.txt requirements.txt
12
 
13
+ RUN pip install --no-cache-dir "numpy<2"
14
+
15
+ RUN pip install --no-cache-dir torch==2.4.0+cpu torchvision==0.19.0+cpu --index-url https://download.pytorch.org/whl/cpu
16
 
17
  RUN pip install --no-cache-dir -r requirements.txt
18
 
requirements.txt CHANGED
@@ -1,10 +1,8 @@
1
  flask==3.1.0
2
- torch>=2.0.0
3
- torchvision>=0.15.0
4
- transformers>=4.30.0
5
  Pillow>=10.0.0
6
  albumentations>=1.3.0
7
- numpy>=1.24.0
8
- opencv-python>=4.8.0
9
  pydicom>=2.4.0
10
  gunicorn>=21.2.0
 
1
  flask==3.1.0
2
+ transformers>=4.40.0
 
 
3
  Pillow>=10.0.0
4
  albumentations>=1.3.0
5
+ numpy<2
6
+ opencv-python-headless>=4.8.0
7
  pydicom>=2.4.0
8
  gunicorn>=21.2.0