Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.6.3-cudnn9-runtime-ubuntu22.04
+FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
@@ -10,11 +10,23 @@ ENV HF_HOME=/tmp/.huggingface
 
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y --no-install-recommends     python3     python3-pip     python3-dev     python3-venv     ffmpeg     git     && ln -sf /usr/bin/python3 /usr/bin/python     && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3 \
+    python3-pip \
+    python3-dev \
+    python3-venv \
+    ffmpeg \
+    git \
+    && ln -sf /usr/bin/python3 /usr/bin/python \
+    && rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt /app/requirements.txt
 
-RUN python -m pip install --upgrade pip==24.3.1 setuptools==75.8.0 wheel==0.45.1 &&     pip install --no-cache-dir       torch==2.8.0 torchaudio==2.8.0 torchvision==0.23.0       --index-url https://download.pytorch.org/whl/cu126 &&     pip install --no-cache-dir -r /app/requirements.txt
+RUN python -m pip install --upgrade pip==24.3.1 setuptools==75.8.0 wheel==0.45.1 && \
+    pip install --no-cache-dir \
+      torch==2.8.0 torchaudio==2.8.0 torchvision==0.23.0 \
+      --index-url https://download.pytorch.org/whl/cu126 && \
+    pip install --no-cache-dir -r /app/requirements.txt
 
 COPY app.py /app/app.py
 COPY README.md /app/README.md