Update Dockerfile

Dockerfile

@@ -11,8 +11,8 @@ COPY requirements.txt .
 
 # Install any needed packages specified in requirements.txt
 # Using --no-cache-dir to reduce image size
-# Specifying a CPU-only version of PyTorch to significantly reduce image size
-RUN pip install --no-cache-dir -r requirements.txt --index-url https://download.pytorch.org/whl/cpu
+# Installs packages from the standard PyPI and also looks in the PyTorch index for torch
+RUN pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu 
 
 # Copy the rest of your application's code into the container
 # This includes app.py, your trained model (.pth), and the scaler (.pkl)