| FROM python:3.10-slim | |
| WORKDIR /app | |
| # Install dependencies | |
| RUN pip install --no-cache-dir \ | |
| fastapi \ | |
| uvicorn[standard] \ | |
| keras \ | |
| huggingface-hub \ | |
| tokenizers \ | |
| transformers \ | |
| tensorflow | |
| # Copy head node code | |
| COPY app.py /app/app.py | |
| # Expose port | |
| EXPOSE 7860 | |
| # Run | |
| CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] |