Update Dockerfile

Dockerfile

@@ -7,9 +7,20 @@ WORKDIR /app
 # Install system dependencies including curl
 RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
 
+# Install Rust and Cargo
+RUN apt-get update && \
+    apt-get install -y curl && \
+    curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && \
+    export PATH="$HOME/.cargo/bin:$PATH" && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Ensure Cargo is in the PATH
+ENV PATH="/root/.cargo/bin:${PATH}"
+
 # Download the model files from Zenodo
-RUN curl -o classification_model.ckpt -L https://zenodo.org/record/11116990/files/classification_model.ckpt
-RUN curl -o segmentation_model.pt -L https://zenodo.org/record/11116990/files/segmentation_model.pt
+#RUN curl -o classification_model.ckpt -L https://zenodo.org/record/11116990/files/classification_model.ckpt
+#RUN curl -o segmentation_model.pt -L https://zenodo.org/record/11116990/files/segmentation_model.pt
 
 # Copy the current directory contents into the container at /app
 COPY . /app