Update Dockerfile

Dockerfile

@@ -3,7 +3,7 @@ FROM python:3.11.5
 RUN useradd -m -u 1000 user
 WORKDIR /app
 COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
 
 COPY --chown=user . /app
 CMD ["python", "/app/UI/main.py"]