diff --git a/app/Dockerfile b/app/Dockerfile
index a2051f5..4b3045b 100644
--- a/app/Dockerfile
+++ b/app/Dockerfile
@@ -5,4 +5,4 @@ RUN apt-get update \
     && apt-get autoremove -y \
     && apt-get clean -y \
     && rm -rf /var/lib/apt/lists/* \
-    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121 --no-cache-dir
\ No newline at end of file
+    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126 --no-cache-dir
\ No newline at end of file