Upload folder using huggingface_hub

Dockerfile

@@ -2,6 +2,14 @@ FROM python:3.9-slim
 
 WORKDIR /code
 
+# Create a directory for model cache with appropriate permissions
+RUN mkdir -p /code/model_cache && chmod -R 777 /code/model_cache
+
+# Set environment variables for model caching
+ENV HF_HOME=/code/model_cache
+ENV TRANSFORMERS_CACHE=/code/model_cache
+ENV HUGGINGFACE_HUB_CACHE=/code/model_cache
+
 # Copy requirements and install
 COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt