Mahdi Hajiabadi commited on
Commit
67f0fa4
·
1 Parent(s): 5ff6c1b

update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -5
Dockerfile CHANGED
@@ -15,14 +15,15 @@ RUN apt-get update && apt-get install -y \
15
  RUN pip install --no-cache-dir --upgrade pip
16
 
17
  # Install dependencies (without the +cpu suffix)
18
- RUN pip install --no-cache-dir \
19
- torch==2.1.0 \
20
- torchvision==0.15.0 \
 
21
  git+https://github.com/huggingface/transformers \
22
  accelerate \
23
- qwen-vl-utils[decord] \
24
  fastapi \
25
- uvicorn[standard]
26
 
27
  # Copy application files
28
  COPY --chown=1000:1000 . /app
 
15
  RUN pip install --no-cache-dir --upgrade pip
16
 
17
  # Install dependencies (without the +cpu suffix)
18
+ RUN pip install --no-cache-dir --upgrade pip && \
19
+ pip install --no-cache-dir \
20
+ torch \
21
+ torchvision \
22
  git+https://github.com/huggingface/transformers \
23
  accelerate \
24
+ qwen-vl-utils[decord]==0.0.8 \
25
  fastapi \
26
+ uvicorn[standard]
27
 
28
  # Copy application files
29
  COPY --chown=1000:1000 . /app