Update Dockerfile

Dockerfile

@@ -8,14 +8,14 @@ RUN chmod 777 /app
 # Set environment variables
 ENV PYTHONUNBUFFERED=1 \
     PYTHONDONTWRITEBYTECODE=1 \
-    TRANSFORMERS_CACHE=/.cache
+    HF_HOME=/.cache
 
 # Set working directory
 WORKDIR /app
 
 # Create a directory for transformers cache
-RUN mkdir -p $TRANSFORMERS_CACHE
-RUN chmod 777 $TRANSFORMERS_CACHE
+RUN mkdir -p $HF_HOME
+RUN chmod 777 $HF_HOME
 
 # Copy requirements.txt to the working directory
 COPY requirements.txt .