hanantonio commited on
Commit
e4b283c
·
verified ·
1 Parent(s): 6b74312

Upload 2 files

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +1 -1
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.21.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==1.11.0 torchvision==0.12.0
26
  RUN pip install fastopic
27
 
28
  # Expose Streamlit port
requirements.txt CHANGED
@@ -20,7 +20,7 @@ nltk==3.8.1
20
 
21
  # Torch (required by fastopic/topmost)
22
  torch==1.11.0
23
- torchvision==0.21.0
24
 
25
  # TensorFlow
26
  tensorflow==2.20.0
 
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