Update Dockerfile

Dockerfile

@@ -1,31 +1,45 @@
-FROM python:3.10-slim
+# Use Python 3.9 base image
+FROM python:3.9-slim
 
-WORKDIR /code
+# Set working directory
+WORKDIR /app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
+    git \
+    wget \
+    curl \
     build-essential \
-    libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
-# Create required directories
-RUN mkdir -p /code/cache && \
-    mkdir -p /code/models && \
-    mkdir -p /code/audio_files && \
-    chmod -R 777 /code
+# Install gdown for Google Drive downloads
+RUN pip install gdown
 
-# Copy requirements first to leverage Docker cache
-COPY ./requirements.txt /code/requirements.txt
+# Copy requirements first to leverage Docker layer caching
+COPY requirements.txt .
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the application
+# Clone the yarngpt repository
+RUN git clone https://github.com/saheedniyi02/yarngpt.git
+
+# Download model files
+RUN wget https://huggingface.co/novateur/WavTokenizer-medium-speech-75token/resolve/main/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml
+RUN gdown 1-ASeEkrn4HY49yZWHTASgfGFNXdVnLTt
+
+# Copy application code
 COPY . .
 
+# Create audio files directory
+RUN mkdir -p audio_files
+
+# Expose port
+EXPOSE 7860
+
 # Set environment variables
-ENV PYTHONPATH=/code
-ENV TRANSFORMERS_CACHE=/code/cache
-ENV HF_HOME=/code/cache
+ENV PYTHONPATH="${PYTHONPATH}:/app"
+ENV HF_HOME="/tmp/huggingface"
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run the application
+CMD ["python", "app.py"]