Update Dockerfile

Dockerfile

@@ -8,11 +8,7 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-# copy everything needed
-COPY requirements.txt ./
-COPY src/ ./src/
-COPY val.csv ./val.csv
-COPY model_trainer/ ./model_trainer/
+COPY . .
 
 RUN pip3 install -r requirements.txt