Update Dockerfile

Dockerfile

@@ -25,7 +25,7 @@ RUN pip install --no-cache-dir numpy==1.26.4 && \
     pip install --no-cache-dir scipy==1.11.4 && \
     pip install --no-cache-dir scikit-learn==1.3.2 && \
     pip install --no-cache-dir matplotlib==3.7.2 && \
-    pip install --no-cache-dir torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118 \
+    pip install --no-cache-dir torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118 && \
     pip install --no-cache-dir -r requirements.txt
 
 # Copy application