Update Dockerfile

Dockerfile

@@ -1,13 +1,28 @@
+# 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 .
 
+# Install the required Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Download the model and tokenizer
+RUN mkdir -p /app/models && \
+    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 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"]