| FROM python:3.10 | |
| WORKDIR /app | |
| COPY requirements.txt . | |
| RUN pip install -r requirements.txt | |
| RUN python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; AutoModelForCausalLM.from_pretrained('facebook/opt-125m'); AutoTokenizer.from_pretrained('facebook/opt-125m')" | |
| COPY . . | |
| EXPOSE 7860 | |
| CMD ["python", "app.py"] |