hanantonio commited on
Commit
9434bda
·
verified ·
1 Parent(s): e4b283c

Upload 2 files

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +2 -2
Dockerfile CHANGED
@@ -22,7 +22,7 @@ RUN pip install --upgrade pip
22
  RUN pip install --no-cache-dir -r requirements.txt
23
 
24
  #install
25
- RUN pip install torch==1.11.0 torchvision==0.12.0
26
  RUN pip install fastopic
27
 
28
  # Expose Streamlit port
 
22
  RUN pip install --no-cache-dir -r requirements.txt
23
 
24
  #install
25
+ RUN pip install torch==2.6.0 torchvision==0.21.0
26
  RUN pip install fastopic
27
 
28
  # Expose Streamlit port
requirements.txt CHANGED
@@ -19,8 +19,8 @@ scikit-learn==1.6.1
19
  nltk==3.8.1
20
 
21
  # Torch (required by fastopic/topmost)
22
- torch==1.11.0
23
- torchvision==0.12.0
24
 
25
  # TensorFlow
26
  tensorflow==2.20.0
 
19
  nltk==3.8.1
20
 
21
  # Torch (required by fastopic/topmost)
22
+ torch==2.6.0
23
+ torchvision==0.21.0
24
 
25
  # TensorFlow
26
  tensorflow==2.20.0