Dockerfile

FROM python:3.10-slim

WORKDIR /app

RUN apt-get update && apt-get install -y git

RUN pip install torch transformers accelerate fastapi uvicorn huggingface_hub

COPY . .

EXPOSE 7860

CMD ["bash", "-c", "python download_model.py && uvicorn app:app --host 0.0.0.0 --port 7860"]