Update Dockerfile

Dockerfile

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