Update Dockerfile

Dockerfile

@@ -1,17 +1,10 @@
-# Use a lightweight Python image
 FROM python:3.10-slim
-
-# Set working directory
 WORKDIR /app
-
-# Set environment variables for Hugging Face cache
 ENV HF_HOME=/tmp/huggingface
 ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
 ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
-
-# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     libgl1-mesa-glx \
@@ -21,31 +14,18 @@ RUN apt-get update && apt-get install -y \
     libxrender-dev \
     wget \
     git \
+    cmake \
+    g++ \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
-
-# Create necessary directories with appropriate permissions
 RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggingface/datasets \
     && chmod -R 777 /tmp/uploads /tmp/results /tmp/huggingface
-
-# Copy requirements file and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -U pip && \
     pip install --no-cache-dir -r requirements.txt
-
-# Download U^2-Net model file from Google Drive
 RUN gdown https://drive.google.com/uc?id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ -O u2net.pth
-
-# Copy tsr module, U^2-Net module, and application code
-COPY tsr /app/tsr
 COPY u2net.py .
 COPY app.py .
-
-# Cache bust to ensure fresh build
 RUN echo "Build $(date)"
-
-# Expose the port the app runs on (standard for Hugging Face Spaces)
 EXPOSE 7860
-
-# Command to run the application
 CMD ["python", "app.py"]