FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime
WORKDIR /app
COPY . .
RUN pip install --no-cache-dir -r requirements.txt \
    && rm /app/requirements.txt
RUN mkdir -p /app/workspace \
    && mkdir -p /app/model
CMD [ "python", "server.py"]