vqa-demo-2 / Dockerfile
duyan2803's picture
Update Dockerfile
e2a9288 verified
raw
history blame contribute delete
608 Bytes
FROM python:3.10
RUN apt-get update && apt-get install -y \
git \
git-lfs \
ffmpeg \
libsm6 \
libxext6 \
cmake \
rsync \
libgl1-mesa-glx \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /home/user/app
RUN pip install --no-cache-dir pip -U
# Cài torch và transformers trước với force-reinstall
RUN pip install --no-cache-dir --force-reinstall torch==2.2.0
RUN pip install --no-cache-dir --force-reinstall transformers==4.38.2
# Cài các gói còn lại
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY app.py .
CMD ["python", "app.py"]