Update Dockerfile

Dockerfile

@@ -1,18 +1,19 @@
-# Use a common, stable Python image used for ML deployments
-FROM python:3.10
+# Use the official PyTorch base image (CPU-only version)
+FROM pytorch/pytorch:2.0.1-cpu
 
 WORKDIR /app
 
-# The necessary build tools are usually included, but we'll ensure common libraries are present
+# Install necessary system packages for Streamlit and Git
 RUN apt-get update && apt-get install -y \
     curl \
     git \
     libffi-dev \
+    libssl-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy and install dependencies
+# Copy and install ONLY non-torch dependencies (torch is already installed)
 COPY requirements.txt ./
-# We use --no-cache-dir to prevent disk space issues, and --force-reinstall to clear old broken installs
+# Install non-torch dependencies, ignoring any torch/cuda mentions
 RUN pip3 install --no-cache-dir --force-reinstall -r requirements.txt
 
 # Copy application source code