Update

Dockerfile

@@ -1,17 +1,27 @@
 FROM python:3.10.9
 
+# Create user with home directory
 RUN useradd -m -u 1000 user
+
+# Switch to the created user
 USER user
-ENV HOME=/home/user
-ENV PATH=/home/user/.local/bin:$PATH
-ENV TRANSFORMERS_CACHE=/home/user/.cache
 
-COPY --chown=user . $HOME/Redaction
+# Set environment variables
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    TRANSFORMERS_CACHE=/home/user/.cache
 
+# Create necessary directories with the right permissions
+RUN mkdir -p $HOME/.cache && chmod -R 777 $HOME/.cache
+
+# Set the working directory
 WORKDIR $HOME/Redaction
 
-RUN mkdir -p $HOME/.cache && chown -R user:user $HOME/.cache
+# Copy project files to the working directory
+COPY --chown=user . $HOME/Redaction
 
+# Install dependencies
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
+# Set the command to run the application
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -1,5 +1,5 @@
 import os
-os.environ["TRANSFORMERS_CACHE"] = "/.cache"
+os.environ["HF_HOME"] = "/.cache"
 
 import re
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
@@ -35,5 +35,5 @@ async def mask_input(query):
     return {"data" : output}
 
 if __name__ == '__main__':
-    os.environ["TRANSFORMERS_CACHE"] = "/.cache"
+    os.environ["HF_HOME"] = "/.cache"
     uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=True, workers=1)