Update Dockerfile

Dockerfile

@@ -1,41 +1,39 @@
 FROM python:3.11-slim
 
 ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
 ENV PORT=7860
 
 WORKDIR /app
 
-# --- System dependencies (OpenCV, ffmpeg, etc.) ---
+# --- System dependencies (Optimized for WebRTC & CV) ---
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
+    pkg-config \
     git \
-    ca-certificates \
     ffmpeg \
-    libgl1 \
-    libglib2.0-0 \
-    libsm6 \
-    libxrender1 \
-    libxext6 \
+    libavdevice-dev libavfilter-dev libavformat-dev libavcodec-dev \
+    libswresample-dev libswscale-dev libavutil-dev \
+    libgl1 libglib2.0-0 \
+    # Speed trick: TCMalloc for better memory management in PyTorch
+    libgoogle-perftools-dev \ 
     && rm -rf /var/lib/apt/lists/*
 
-# --- Copy project ---
-COPY . /app
+# Set TCMalloc as the memory allocator
+ENV LD_PRELOAD="/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4"
 
-# --- Create required folders (uploads, results, Model) ---
-RUN mkdir -p /app/static/uploads /app/static/results 
+# --- Python Setup ---
+RUN python -m pip install --upgrade pip setuptools wheel
 
-# --- Give write permissions (HF Spaces safe) ---
-RUN chmod -R 777 /app/static/uploads /app/static/results 
+# --- Dependencies ---
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-# --- Python tooling ---
-RUN python -m pip install --upgrade pip setuptools wheel huggingface_hub
+COPY . .
 
-# --- Install dependencies ---
-COPY requirements.txt /app/requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+# Setup permissions
+RUN mkdir -p static/uploads static/results && chmod -R 777 static/uploads static/results
 
-# --- HF Spaces port ---
 EXPOSE 7860
 
-# --- Run Flask-SocketIO app ---
 CMD ["python", "-u", "app.py"]