Update Dockerfile

Dockerfile

@@ -1,28 +1,26 @@
-# Use the official Python 3.9 slim image as the base image
 FROM python:3.9-slim
 
-# Set the working directory inside the container
 WORKDIR /app
 
-# Copy the requirements file into the container
-COPY requirements.txt .
+# Create and configure cache directory
+RUN mkdir -p /app/.cache && chmod 777 /app/.cache
+ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HF_HOME=/app/.cache
 
-# Install the required Python dependencies
+# Install dependencies
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Download the model and tokenizer
-RUN mkdir -p /app/models && \
+# 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; \
                tokenizer = AutoTokenizer.from_pretrained('suno/bark'); \
                processor = AutoProcessor.from_pretrained('suno/bark'); \
                tokenizer.save_pretrained('/app/models/suno-bark'); \
                processor.save_pretrained('/app/models/suno-bark')"
 
+# Copy application code
 COPY app.py .
 
-ENV TRANSFORMERS_CACHE=/app/.cache
-ENV HF_HOME=/app/.cache
-
 EXPOSE 7860
-
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]