Update Dockerfile

Dockerfile

@@ -1,29 +1,43 @@
-FROM python:3.9-slim
+FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
 
-WORKDIR /app
+WORKDIR /code
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
+    wget \
+    ffmpeg \
+    libsndfile1 \
     build-essential \
+    python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first to leverage Docker cache
+# Create cache directory with proper permissions
+RUN mkdir -p /.cache/huggingface && chmod -R 777 /.cache
+
+# Clean pip cache and install dependencies
+RUN pip cache purge
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
-COPY . .
-
 # Clone and install IndicTransToolkit
 RUN git clone https://github.com/VarunGumma/IndicTransToolkit \
     && cd IndicTransToolkit \
     && pip install --editable ./
 
-# Create necessary directories
-RUN mkdir -p templates
+# Copy application code
+COPY . .
+
+# Create directories and set permissions
+RUN mkdir -p templates static/audio \
+    && chmod -R 777 static/audio
+
+# Set environment variables
+ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:32
+ENV TRANSFORMERS_CACHE=/.cache/huggingface/hub
+ENV HF_HOME=/.cache/huggingface
 
-# Expose the port the app runs on
+# Expose port
 EXPOSE 7860
 
 # Command to run the application