Update Dockerfile

Dockerfile

@@ -1,13 +1,13 @@
-FROM python:3.11.5
+FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-runtime
 
 RUN useradd -m -u 1000 user
+
 WORKDIR /app
 COPY --chown=user ./requirements.txt requirements.txt
 RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
-RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
 RUN pip install gradio opencv-python pandas ultralytics
 
 COPY --chown=user . /app
 
 EXPOSE 1071
-CMD ["python", "/app/UI/Main.py"]
+CMD ["python", "/app/UI/Main.py"]