Update Dockerfile

Dockerfile

@@ -1,80 +1,37 @@
-FROM python:3.10-slim
+FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install system dependencies for AI models and image processing
-RUN apt-get update && apt-get install -y \
-    gcc \
-    g++ \
-    git \
-    curl \
-    wget \
+# Install minimal system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1 \
     libglib2.0-0 \
-    libsm6 \
-    libxext6 \
-    libxrender-dev \
-    libgl1-mesa-glx \
-    ffmpeg \
-    && rm -rf /var/lib/apt/lists/* \
-    && apt-get clean
+    && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first for better Docker layer caching
+# Copy requirements
 COPY requirements.txt .
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Create necessary directories
-RUN mkdir -p \
-    /app/models \
-    /app/temp \
-    /app/uploads \
-    /app/outputs \
-    /app/examples
+# Create cache directory
+RUN mkdir -p /app/cache
 
-# Copy application code
-COPY . .
-
-# Pre-download and cache AI models during build
-RUN python -c "
-import rembg
-import torch
-from transformers import pipeline
-
-# Pre-download background removal model
-print('Downloading background removal model...')
-rembg.remove((255, 255, 255, 255))
-
-# Pre-download other essential models
-print('Pre-caching AI models...')
-try:
-    # Face detection model
-    from insightface.app import FaceAnalysis
-    print('Models pre-cached successfully')
-except Exception as e:
-    print(f'Model caching: {e}')
-" || true
-
-# Create non-root user for security
-RUN groupadd -r appuser && useradd -r -g appuser appuser
-RUN chown -R appuser:appuser /app
-USER appuser
-
-# Expose ports
-EXPOSE 7860  # Gradio
-EXPOSE 8000  # FastAPI
-
-# Health check
-HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
-    CMD python -c "import requests; requests.get('http://localhost:7860/', timeout=10)" || exit 1
+# Copy application
+COPY app.py .
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV GRADIO_SERVER_PORT=7860
-ENV MODEL_CACHE_DIR="/app/models"
-ENV TEMP_DIR="/app/temp"
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HUGGINGFACE_HUB_CACHE=/app/cache
+
+# Expose port
+EXPOSE 7860
+
+# Run as non-root user
+RUN useradd -m appuser && chown -R appuser:appuser /app
+USER appuser
 
-# Run the application
 CMD ["python", "app.py"]