Update Dockerfile

Dockerfile

@@ -1,9 +1,13 @@
+# Use the official Python 3.9 slim image
 FROM python:3.9-slim
 
+# Set the working directory
 WORKDIR /app
 
-# Create and configure cache directory
-RUN mkdir -p /app/.cache && chmod 777 /app/.cache
+# Create and configure the cache directory with recursive permissions
+RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
+
+# Set environment variables for caching
 ENV TRANSFORMERS_CACHE=/app/.cache
 ENV HF_HOME=/app/.cache
 
@@ -13,11 +17,13 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 # Pre-download the model (replace 'suno/bark' with a valid model if needed)
 RUN mkdir -p /app/models/suno-bark && \
-    python -c "from transformers import AutoTokenizer, AutoProcessor; \
+    python -c "from transformers import AutoTokenizer, AutoProcessor, BarkModel; \
                tokenizer = AutoTokenizer.from_pretrained('suno/bark'); \
                processor = AutoProcessor.from_pretrained('suno/bark'); \
+               model = BarkModel.from_pretrained('suno/bark'); \
                tokenizer.save_pretrained('/app/models/suno-bark'); \
-               processor.save_pretrained('/app/models/suno-bark')"
+               processor.save_pretrained('/app/models/suno-bark'); \
+               model.save_pretrained('/app/models/suno-bark')"
 
 # Copy application code
 COPY app.py .