Update Dockerfile

Dockerfile

@@ -1,13 +1,11 @@
-# Base image
 FROM python:3.11-slim
 
-# Set working directory
 WORKDIR /app
 
-# -------------------------------------------------
-# System dependencies (TensorFlow / HDF5 support)
-# -------------------------------------------------
+# --- System deps ---
 RUN apt-get update && apt-get install -y \
+    git \
+    git-lfs \
     libsm6 \
     libxext6 \
     libxrender-dev \
@@ -15,41 +13,20 @@ RUN apt-get update && apt-get install -y \
     pkg-config \
     && rm -rf /var/lib/apt/lists/*
 
-# -------------------------------------------------
-# Python dependencies
-# -------------------------------------------------
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+# --- Enable Git LFS ---
+RUN git lfs install
 
-# -------------------------------------------------
-# Create required directories FIRST
-# -------------------------------------------------
-RUN mkdir -p models logs
-
-# -------------------------------------------------
-# Copy application code
-# -------------------------------------------------
-COPY main.py .
-COPY schemas.py .
-COPY inference/ ./inference/
-
-# -------------------------------------------------
-# Copy trained model (FIXED)
-# -------------------------------------------------
-COPY models/Forest_Segmentation_Best.keras models/Forest_Segmentation_Best.keras
-
-# -------------------------------------------------
-# Environment variables
-# -------------------------------------------------
-ENV PYTHONUNBUFFERED=1
-ENV PORT=7860
+# --- Copy repo ---
+COPY . .
+
+# --- Pull LFS files INSIDE container ---
+RUN git lfs pull
 
-# -------------------------------------------------
-# Hugging Face Spaces port
-# -------------------------------------------------
+# --- Install Python deps ---
+RUN pip install --no-cache-dir -r requirements.txt
+
+# --- Runtime ---
 EXPOSE 7860
+ENV PYTHONUNBUFFERED=1
 
-# -------------------------------------------------
-# Run FastAPI server
-# -------------------------------------------------
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]