Update Dockerfile

Dockerfile

@@ -1,10 +1,8 @@
-# Use official Python runtime as base image
 FROM python:3.11-slim
 
-# Set working directory in container
 WORKDIR /app
 
-# Install system dependencies (OpenCV, HDF5 support)
+# System dependencies for TensorFlow / HDF5
 RUN apt-get update && apt-get install -y \
     libsm6 \
     libxext6 \
@@ -13,35 +11,25 @@ RUN apt-get update && apt-get install -y \
     pkg-config \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements file
-COPY requirements.txt .
-
 # Install Python dependencies
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY main.py .
-COPY model.py .
 COPY schemas.py .
 COPY inference/ ./inference/
 
-# Copy pre-trained model (use .keras format only)
+# Copy model
 COPY models/Forest_Segmentation_Best.keras ./models/
 
-# Create logs directory
+# Logs directory
 RUN mkdir -p logs
 
-# Expose port (Hugging Face Spaces default)
+# Hugging Face uses 7860
 EXPOSE 7860
 
-# Set environment for production
 ENV PYTHONUNBUFFERED=1
-ENV PORT=7860
-ENV HOST=0.0.0.0
-
-# Health check
-HEALTHCHECK --interval=30s --timeout=10s --start-period=40s --retries=3 \
-    CMD python -c "import requests; requests.get('http://localhost:7860/health')" || exit 1
 
-# Run FastAPI application with uvicorn
+# Start FastAPI
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]