From b513a4eea27fd10d16d535ea3a3425966e559e86 Mon Sep 17 00:00:00 2001
From: muellerdo <domey038@live.de>
Date: Tue, 23 Jan 2024 22:20:41 +0100
Subject: [PATCH] feat: updated Dockerfile

---
 Dockerfile | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 65982cc..c09e432 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -44,10 +44,10 @@ RUN apt-get update && \
     apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
 
 # Update Python pip
-RUN python -m pip install pip --upgrade
+RUN python3.10 -m pip install pip --upgrade
 
 # Install DeepGleason from local git repo
-RUN pip install -r /root/DeepGleason/requirements.txt
+RUN python3.10 -m pip install -r /root/DeepGleason/requirements.txt
 
 # Create working directory
 VOLUME ["/data"]
@@ -56,4 +56,4 @@ WORKDIR "/root/DeepGleason/"
 #-----------------------------------------------------#
 #                       Startup                       #
 #-----------------------------------------------------#
-ENTRYPOINT ["code/main.py"]
+ENTRYPOINT ["python3.10 code/main.py --input /data/ --output /data/ --model models/model.ConvNeXtBase --predictions /data/predictions.csv"]