FROM python:3.10

# Install system libraries
RUN apt-get update && apt-get install -y libgl1 git-lfs

# Copy your app files
COPY . /app
WORKDIR /app

# Pull LFS files
RUN cd NAFNet && git lfs install && git lfs pull

# Install Python dependencies
RUN pip install -r requirements.txt

CMD ["python", "NAFNetModel/app.py"]