Update Dockerfile

Dockerfile

@@ -1,30 +1,33 @@
+# Use Python 3.10 with slim image
 FROM python:3.10-slim
 
+# Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
     HF_HOME=/app/.cache \
-    TRANSFORMERS_CACHE=/app/.cache
+    TRANSFORMERS_VERBOSITY=error  # Suppress warnings
 
-# Create directories and set permissions
+# Create directories with proper permissions
 WORKDIR /app
 RUN mkdir -p /app/.cache /app/models && \
     chmod -R 777 /app/.cache /app/models
 
-# Install system dependencies for audio processing
+# Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    libsndfile1 \
-    ffmpeg && \
+    libsndfile1 ffmpeg && \
     rm -rf /var/lib/apt/lists/*
 
 # Install Python dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -U pip && \
+    pip install --no-cache-dir -r requirements.txt
 
-# Pre-download and cache models
+# Pre-download VALID model (replace with actual model name)
 RUN 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'); \
+               model_name = 'suno/bark-small'; \
+               tokenizer = AutoTokenizer.from_pretrained(model_name); \
+               processor = AutoProcessor.from_pretrained(model_name); \
+               model = BarkModel.from_pretrained(model_name); \
                tokenizer.save_pretrained('/app/models/suno-bark'); \
                processor.save_pretrained('/app/models/suno-bark'); \
                model.save_pretrained('/app/models/suno-bark')"