Update Dockerfile

Dockerfile

@@ -3,17 +3,22 @@ FROM python:3.9-slim
 WORKDIR /app
 
 RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    git \
     ffmpeg \
     libsndfile1 \
-    git \
-    curl && rm -rf /var/lib/apt/lists/*
+    && rm -rf /var/lib/apt/lists/*
+
+RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
+
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+ENV HF_HOME=/app/.cache/huggingface
+ENV TORCH_HOME=/app/.cache/torch
 
 COPY requirements.txt ./
 COPY src/ ./src/
 
-RUN mkdir /app/cache && chmod 777 /app/cache
-ENV TRANSFORMERS_CACHE=/app/cache
-
 RUN pip install --no-cache-dir -r requirements.txt
 
 EXPOSE 8501