Refactor

Dockerfile

@@ -8,8 +8,11 @@ WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-# Set the TRANSFORMERS_CACHE environment variable
-ENV TRANSFORMERS_CACHE /code/.cache/huggingface/hub
+# Set the HF_HOME environment variable
+ENV HF_HOME /code/.cache/huggingface
+
+# Create cache directory with appropriate permissions
+RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt