Update Dockerfile

Dockerfile

@@ -1,37 +1,35 @@
-FROM python:3.9-slim
+FROM python:3.10-slim
 
-# System dependencies required for Python packages
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential \
-    gcc \
-    curl \
-    && rm -rf /var/lib/apt/lists/*
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+ENV OMP_NUM_THREADS=1
+ENV MKL_NUM_THREADS=1
+ENV OPENBLAS_NUM_THREADS=1
+ENV VECLIB_MAXIMUM_THREADS=1
+ENV NUMEXPR_NUM_THREADS=1
 
-# Set working directory
 WORKDIR /app
 
-# Install blis and googletrans separately to avoid building from source
-RUN pip install --no-cache-dir --prefer-binary \
-    blis==0.7.9 \
-    googletrans==4.0.0rc1
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    git \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements and install remaining dependencies
-COPY requirements.txt .
-RUN pip install --no-cache-dir --prefer-binary -r requirements.txt
+# Copy requirements first for better caching
+COPY requirements.txt /app/
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Download spaCy model
-RUN python -m spacy download en_core_web_sm
+# Copy the rest of the application
+COPY . /app
 
-# Copy your application
-COPY . .
+# Set environment variables for PyTorch
+ENV TORCH_CUDA_VERSION=cu118
+ENV TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
+ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512
 
-# Expose required port
 EXPOSE 7860
 
-# Flask environment config
-ENV FLASK_APP=app.py
-ENV FLASK_RUN_PORT=7860
-ENV FLASK_RUN_HOST=0.0.0.0
-
-# Run the Flask server
-CMD ["flask", "run"]
+# Run with multiple workers for better CPU utilization
+CMD ["python", "app.py"]