Update Dockerfile

Dockerfile

@@ -1,10 +1,17 @@
+# 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 \
@@ -12,20 +19,32 @@ RUN apt-get update && apt-get install -y \
     libsm6 \
     libxext6 \
     libxrender-dev \
-    wget \
     git \
-    cmake \
-    g++ \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
+
+# Install additional dependencies for PyMeshLab
+RUN apt-get update && apt-get install -y \
+    libeigen3-dev \
+    libgmp-dev \
+    libmpfr-dev \
+    && 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
-RUN gdown https://drive.google.com/uc?id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ -O u2net.pth
-COPY u2net.py .
+
+# Copy application code
 COPY app.py .
-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"]