|
|
FROM python:3.10-slim |
|
|
|
|
|
WORKDIR /app |
|
|
|
|
|
|
|
|
COPY requirements.txt . |
|
|
RUN pip install --no-cache-dir -r requirements.txt |
|
|
|
|
|
|
|
|
RUN mkdir -p /app/cache |
|
|
RUN chmod -R 777 /app/cache |
|
|
ENV TRANSFORMERS_CACHE=/app/cache |
|
|
|
|
|
|
|
|
RUN find /app/cache -name '*.lock' -type f -delete |
|
|
|
|
|
|
|
|
RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; AutoTokenizer.from_pretrained('microsoft/BioGPT-Large'); AutoModelForCausalLM.from_pretrained('microsoft/BioGPT-Large')" |
|
|
|
|
|
|
|
|
COPY . . |
|
|
|
|
|
|
|
|
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] |
|
|
|
