Update Dockerfile

Dockerfile

@@ -31,11 +31,9 @@ WORKDIR /app
 # Copy requirements first for better caching
 COPY requirements.txt .
 
-# Install PyTorch CPU version first
-RUN pip3 install --no-cache-dir torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu
-
-# Install other Python dependencies
-RUN pip3 install --no-cache-dir -r requirements.txt
+# Install Python dependencies
+RUN pip3 install --no-cache-dir --upgrade pip && \
+    pip3 install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY app.py .