Update Dockerfile

Dockerfile

@@ -1,49 +1,52 @@
-# Use official Python image
-FROM python:3.10-slim
-
-# Disable interactive prompts
-ENV DEBIAN_FRONTEND=noninteractive
-
-# Set working directory
-WORKDIR /app
-
-# Install system dependencies
-RUN apt-get update && apt-get install -y \
-    ffmpeg \
-    libsm6 \
-    libxext6 \
-    git \
-    curl \
-    && rm -rf /var/lib/apt/lists/*
-
-# Copy requirements and install Python packages
-COPY requirements.txt .
-
-# Upgrade pip first and set timeout + retries
-RUN pip install --upgrade pip \
- && pip config set global.timeout 100 \
- && pip config set global.retries 10 \
- && pip install --no-cache-dir -r requirements.txt
-
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Set Hugging Face cache directory
-ENV HF_HOME=/models
-ENV TRANSFORMERS_CACHE=/models/transformers
-ENV HF_DATASETS_CACHE=/models/datasets
-ENV HF_METRICS_CACHE=/models/metrics
-
-# Pre-download the SAM model from Hugging Face
-RUN python -c "\
-from transformers import SamModel, SamProcessor; \
-SamModel.from_pretrained('Zigeng/SlimSAM-uniform-50'); \
-SamProcessor.from_pretrained('Zigeng/SlimSAM-uniform-50')"
-
-# Copy all project files
-COPY . .
-
-# Expose port for Flask
-EXPOSE 6000
-
-# Run using gunicorn
-CMD ["gunicorn", "--bind", "0.0.0.0:6000", "app:app"]
+# Use official Python image
+FROM python:3.10-slim
+
+# Disable interactive prompts
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Set working directory
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements and install Python packages
+COPY requirements.txt .
+
+# Upgrade pip first and set timeout + retries
+RUN pip install --upgrade pip \
+ && pip config set global.timeout 100 \
+ && pip config set global.retries 10 \
+ && pip install --no-cache-dir -r requirements.txt
+
+RUN mkdir -p /app/static/uploads /app/static/outputs
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+
+# Set Hugging Face cache directory
+ENV HF_HOME=/models
+ENV TRANSFORMERS_CACHE=/models/transformers
+ENV HF_DATASETS_CACHE=/models/datasets
+ENV HF_METRICS_CACHE=/models/metrics
+
+# Pre-download the SAM model from Hugging Face
+RUN python -c "\
+from transformers import SamModel, SamProcessor; \
+SamModel.from_pretrained('Zigeng/SlimSAM-uniform-50'); \
+SamProcessor.from_pretrained('Zigeng/SlimSAM-uniform-50')"
+
+# Copy all project files
+COPY . .
+
+# Expose port for Flask
+EXPOSE 6000
+
+# Run using gunicorn
+CMD ["gunicorn", "--bind", "0.0.0.0:6000", "app:app"]