Update Dockerfile

Dockerfile

@@ -1,21 +1,32 @@
-FROM python:3.9-slim
-
-WORKDIR /app
+# Base image with PyTorch and CUDA (adjust to CPU image if needed)
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
 
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
+    libgl1 \
+    libglib2.0-0 \
+    ffmpeg \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt ./
-COPY src/ ./src/
+# Set working directory
+WORKDIR /app
+
+# Copy your app code and model
+COPY app.py .
+COPY requirements.txt .
+COPY nnunet_model/ /app/nnunet_model/
 
-RUN pip3 install -r requirements.txt
+# Install Python dependencies
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
 
-EXPOSE 8501
+# Clone nnU-Net v2 and install it properly
+RUN git clone -b nnunetv2 https://github.com/MIC-DKFZ/nnUNet.git
+RUN pip install -e ./nnUNet
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Expose port
+EXPOSE 7860
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Launch app
+CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]