Update Dockerfile

Dockerfile

@@ -1,43 +1,36 @@
 # Dockerfile
-FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
+FROM python:3.10-slim
 
-# Jika Anda tidak menggunakan GPU, ganti base image menjadi python:3.10-slim
-# FROM python:3.10-slim
-
-ENV DEBIAN_FRONTEND=noninteractive
+# Set working directory
 WORKDIR /app
 
-# system deps
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential \
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    libgl1 \
+    libglib2.0-0 \
     git \
     wget \
-    curl \
-    ca-certificates \
-    libglib2.0-0 \
-    libsm6 \
-    libxext6 \
-    libxrender-dev \
-    ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
-# copy requirements
-COPY requirements.txt /app/
+# Copy requirements first for better caching
+COPY requirements.txt .
 
-# install python deps
-RUN python -m pip install --upgrade pip setuptools wheel
+# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# copy app
-COPY . /app
+# Copy application files
+COPY app.py .
 
-# make dirs
-RUN mkdir -p uploads results models && chmod -R 777 uploads results models
-
-# copy SAM checkpoint if you have local copy (optional)
-# If you have local model file at build time, uncomment:
-# COPY models/sam_vit_b_01ec64.pth /app/models/sam_vit_b_01ec64.pth
+# Create necessary directories with permissions
+RUN mkdir -p uploads processed models && \
+    chmod -R 777 uploads processed models
 
+# Expose port
 EXPOSE 7860
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
+    CMD python -c "import requests; requests.get('http://localhost:7860/')"
+
+# Run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]