fffiloni commited on
Commit
b7bfd7a
·
verified ·
1 Parent(s): a2af212

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -97,7 +97,8 @@ ENV PYTHONPATH=$HOME/app:$HOME/app/third_party:$PYTHONPATH
97
  RUN python -c "import sys; print('\n'.join(sys.path))"
98
 
99
  RUN pip install --no-cache-dir pybind11>=2.12 && \
100
- pip install --no-cache-dir numpy==1.26.4
 
101
 
102
  # Verify CSD module can be imported (try different methods)
103
  RUN python -c "from third_party.CSD import model; print('CSD model successfully imported')" || \
 
97
  RUN python -c "import sys; print('\n'.join(sys.path))"
98
 
99
  RUN pip install --no-cache-dir pybind11>=2.12 && \
100
+ pip install --no-cache-dir numpy==1.26.4 && \
101
+ pip install --no-cache-dir transformers==4.31.0
102
 
103
  # Verify CSD module can be imported (try different methods)
104
  RUN python -c "from third_party.CSD import model; print('CSD model successfully imported')" || \