Fix torchvision version

Dockerfile

@@ -29,7 +29,7 @@ RUN apt-get -qq update && DEBIAN_FRONTEND=noninteractive apt-get -y install \
 COPY ./requirements.txt ./
 RUN pip install -r ./requirements.txt && \
     # Install torch and torchvision CPU only versions
-    pip install torch==2.2.1+cpu torchvision==0.9.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html && \
+    pip install torch==2.2.1+cpu torchvision==0.17.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html && \
     #Explicitly install ultralytics without optional dependencies (like CUDA). (see https://github.com/ultralytics/ultralytics/issues/749)
     pip install ultralytics==8.1.19 --no-deps && \
     pip cache purge