Dockerfile

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.whl filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,57 @@
+FROM python:3.10
+
+# Set environment variables to force Hugging Face to use a fresh cache directory
+ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+ENV HF_HOME=/tmp/huggingface_cache
+
+# Clean Hugging Face cache
+RUN rm -rf ~/.cache/huggingface/transformers
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y git wget git-lfs && git lfs install && apt-get clean
+
+# Create a non-root user
+RUN useradd -m -u 1000 user
+
+# Create directory for repo and give access to non-root user
+RUN mkdir /ModerationModelCode && chown -R user:user /ModerationModelCode
+
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+# Clone the repo with LFS files
+USER root
+ARG CACHEBUST=4
+RUN GIT_LFS_SKIP_SMUDGE=0 git clone --depth 1 https://huggingface.co/InfosysEnterprise/ModerationModelCode /ModerationModelCode
+RUN git config --global --add safe.directory /ModerationModelCode
+RUN cd /Responsible-Ai-Moderation-Models && git log -1 --oneline && rm -rf .git
+
+USER user
+
+# Check models directory and files
+RUN ls -lR /ModerationModelCode/models
+RUN find /ModerationModelCode/models -type f -exec ls -lh {} +
+
+# Copy local files to Docker (if needed)
+COPY --chown=user . /ModerationModelCode
+
+WORKDIR /ModerationModelCode
+
+# Install .whl files
+RUN pip install --no-cache-dir /ModerationModelCode/lib/privacy-2.0.8-py3-none-any.whl
+RUN pip install --no-cache-dir /ModerationModelCode/lib/en_core_web_lg-3.5.0-py3-none-any.whl
+
+# Install PyTorch and Transformers
+RUN pip install torch==2.2.0 torchvision==0.17.0 --index-url https://download.pytorch.org/whl/cpu
+RUN pip install transformers==4.48.0 diffusers==0.27.2
+
+# Install additional requirements
+RUN pip install --user --no-cache-dir -r requirements/requirement.txt
+
+WORKDIR /ModerationModelCode/src
+
+# Expose Hugging Face Spaces port
+EXPOSE 7860
+
+# Run main app
+CMD ["python", "main.py"]