Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
-# Dockerfile
+# Dockerfile for AI Background Changer
+
 FROM python:3.10-slim
 
-# Set working directory
 WORKDIR /app
 
 # Install system dependencies
@@ -12,25 +12,25 @@ RUN apt-get update && apt-get install -y \
     wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first for better caching
+# Copy requirements first
 COPY requirements.txt .
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application files
-COPY app.py .
+COPY . .
 
 # Create necessary directories with permissions
-RUN mkdir -p uploads processed models && \
-    chmod -R 777 uploads processed models
+RUN mkdir -p uploads results models && chmod -R 777 uploads results models
+
+# Download SAM model if not present (optional - you have it locally)
+# RUN if [ ! -f models/sam_vit_b_01ec64.pth ]; then \
+#     wget -P models/ https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth; \
+#     fi
 
 # Expose port
 EXPOSE 7860
 
-# 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
+# Run application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]