Update Dockerfile

Dockerfile

@@ -53,7 +53,7 @@ WORKDIR $HOME/app
 # Upgrade pip and install numpy<2 first, then PyTorch
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir "numpy<2" && \
-    pip install --no-cache-dir torch==1.7.1 torchvision --extra-index-url https://download.pytorch.org/whl/cu113 && \
+    pip install --no-cache-dir torch==1.8.0 torchvision --extra-index-url https://download.pytorch.org/whl/cu113 && \
     python -c "import torch; print('PyTorch version:', torch.__version__)"
 
 # Clone the repository