Update Dockerfile

Dockerfile

@@ -38,6 +38,8 @@ RUN apt-get update && apt-get install -y \
 #     libcudnn8-dev=${CUDNN_VERSION}.*-1+cuda${CUDA_VERSION} \
 #     && rm -rf /var/lib/apt/lists/*
 # RUN 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
+#
+# If you uncommented the CUDA/PyTorch installation above, set this to 'true'.
 # ENV USE_GPU=true
 # --- END OPTIONAL CUDA/GPU Installation ---
 
@@ -53,10 +55,9 @@ COPY app.py .
 # Expose the port Flask runs on
 EXPOSE 5000
 
-# Set an environment variable for GPU usage.
-# If you uncommented the CUDA installation above, USE_GPU should be true.
-# Otherwise, for CPU-only, keep it false.
-ENV USE_GPU=false # Set to 'true' if you uncommented the CUDA/PyTorch installation above
+# Set USE_GPU environment variable for CPU-only deployment.
+# If you enable the OPTIONAL CUDA/GPU section above, ensure ENV USE_GPU=true is set there.
+ENV USE_GPU=false
 
 # Command to run the Flask application using gunicorn for production serving.
 CMD ["gunicorn", "--bind", "0.0.0.0:5000", "app:app"]