Upload 2 files

Dockerfile

@@ -0,0 +1,49 @@
+# Use an official Python runtime as a parent image
+FROM python:3.10-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Install necessary system dependencies
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    unzip \
+    git \
+    # Add potential ITK/build dependencies (might need adjustment)
+    build-essential \
+    cmake \
+    libgl1 \
+    libglib2.0-0 \
+    # libitk5-dev # Removed: Rely on pip install of itk-elastix
+    && \
+    rm -rf /var/lib/apt/lists/*
+
+# Copy the requirements file first to leverage Docker cache
+COPY requirements.txt .
+
+# Install Python packages specified in requirements.txt
+# Using --no-cache-dir can reduce image size
+RUN pip install --no-cache-dir -r requirements.txt
+
+# --- HD-BET is now copied locally via src/BrainIAC ---
+
+# Copy the rest of the application code (including local HD_BET)
+COPY src/BrainIAC /app/BrainIAC
+
+# Copy static files (like images)
+COPY src/BrainIAC/static /app/BrainIAC/static
+
+# Copy the MNI templates and parameter files
+COPY src/BrainIAC/golden_image /app/BrainIAC/golden_image
+
+# Copy the HD-BET models
+COPY src/BrainIAC/hdbet_model /app/BrainIAC/hdbet_model
+
+# Copy the MCI classification model checkpoint
+COPY src/BrainIAC/checkpoints/mci_model.pt /app/BrainIAC/checkpoints/mci_model.pt
+
+# Make port 7860 available (for Gradio)
+EXPOSE 7860
+
+# Run app_gradio.py when the container launches using python
+CMD ["python", "/app/BrainIAC/app_gradio.py"] 

requirements.txt

@@ -0,0 +1,24 @@
+pytorch-lightning==2.3.3
+monai==1.3.2
+nibabel==5.2.1
+scikit-image==0.21.0
+scikit-learn==1.2.2
+scipy==1.10.1
+seaborn==0.12.2
+numpy==1.23.5
+autograd==1.7.0
+matplotlib==3.7.1
+SimpleITK==2.4.0
+tqdm
+pydicom
+wandb
+lifelines
+torch==2.6.0
+opencv-python
+pandas 
+Flask
+gunicorn
+PyYAML
+dicom2nifti
+itk-elastix
+gradio