Update Dockerfile

Dockerfile

@@ -4,18 +4,15 @@ FROM python:3.10-slim
 # Set the working directory
 WORKDIR /app
 
-# Set environment variables
-ENV TRANSFORMERS_CACHE=/app/cache
-
 # Upgrade pip to the latest version
 RUN pip install --upgrade pip
 
-# Create cache directory
-RUN mkdir -p /app/cache
-
 # Copy the current directory contents into the container at /app
 COPY . /app
 
+# Set the HF_HOME environment variable
+ENV HF_HOME /app/cache
+
 # Install any needed packages specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt