Try fixing TRANSFORMER_CACHE issue #2

Dockerfile

@@ -14,7 +14,8 @@ EXPOSE 7860 8080
 COPY start.sh .
 RUN chmod +x start.sh
 
-RUN mkdir -p /root/.cache/huggingface/hub
-RUN chmod 777 /root/.cache/huggingface/hub
+RUN chmod 777 /workspace
+RUN mkdir -p /workspace/.cache/huggingface/hub
+ENV TRANSFORMERS_CACHE=/workspace/.cache/huggingface/hub
 
 CMD ["./start.sh"]

README.md

@@ -73,7 +73,7 @@ docker build -t gistillery:latest .
 Next run the container:
 
 ```sh
-docker run -p 7860:7860 -p 8080:8080 -e GRADIO_SERVER_NAME=0.0.0.0 -v $HOME/.cache/huggingface/hub:/root/.cache/huggingface/hub gistillery:latest
+docker run -p 7860:7860 -p 8080:8080 -e GRADIO_SERVER_NAME=0.0.0.0 -v $HOME/.cache/huggingface/hub:/workspace/.cache/huggingface/hub gistillery:latest
 ```
 
 Note that the Hugging Face cache folder is mounted as a docker volume to make use of potentially available local model cache instead of downloading the transformers models each time the container is started. To prevent that, remove the `-v ...` parameter. The database used for storing the results is ephemeral and will be deleted when the docker container is stopped. The backend server is also exposed directly via port 8080 to enable DB backups (see below).