| FROM python:3.9-slim | |
| WORKDIR / | |
| # Install git | |
| RUN apt-get update && apt-get install -y git | |
| # Install Hugging face Dependencies | |
| RUN pip install huggingface_hub transformers git+https://github.com/huggingface/transformers.git | |
| # Alternatively, if you want to use git to clone the model repo | |
| RUN git clone https://huggingface.co/InfosysResponsibleAiToolKit/Responsible-Ai-Moderation-Models | |
| WORKDIR /Responsible-Ai-Moderation-Models | |
| RUN ls -l | |
| RUN pip install --no-cache-dir --upgrade -r requirements/requirement.txt | |
| WORKDIR /src | |
| # command | |
| CMD ["python", "main.py"] | |