FINAL FIX: Updated Dockerfile for explicit PyTorch/Accelerate installation and trimmed requirements.txt

Dockerfile

@@ -1,4 +1,4 @@
-# Use a Debian-based Python image. This is often more stable for PyTorch than 'slim'.
+# Use a Debian-based Python image. This is robust for complex scientific libraries.
 FROM python:3.10
 
 # Set the working directory
@@ -7,28 +7,30 @@ WORKDIR /app
 # Expose the port Streamlit runs on
 EXPOSE 8501
 
-# Copy the requirements file and install dependencies
+# Copy the requirements file and application code
 COPY requirements.txt .
 COPY main.py .
 COPY merged.py .
 COPY app.py .
 
-# --- NEW INSTALLATION STEPS FOR ROBUST PYTORCH ---
+# --- ROBUST DEPENDENCY INSTALLATION ---
 
 # 1. Upgrade pip for stability
 RUN pip install --upgrade pip
 
-# 2. Install PyTorch explicitly for CPU (as configured in merged.py)
-# This uses the specific CPU index URL, which is the most reliable way 
-# to ensure the CPU version of PyTorch is installed without relying on 
-# complex dependencies that might not be on the base image.
-RUN pip install torch --index-url https://download.pytorch.org/whl/cpu
+# 2. Install complex, device-dependent packages explicitly for CPU.
+# This ensures that PyTorch, Accelerate, and core Hugging Face libraries
+# are correctly linked and installed before other dependencies.
+RUN pip install \
+    torch --index-url https://download.pytorch.org/whl/cpu \
+    accelerate \
+    transformers \
+    huggingface-hub
 
-# 3. Install the rest of the dependencies
-# The original requirements.txt (excluding torch)
+# 3. Install the remaining dependencies from the trimmed requirements.txt
 RUN pip install -r requirements.txt
 
-# --- END NEW INSTALLATION STEPS ---
+# --- END INSTALLATION ---
 
 # Command to run the Streamlit application
 CMD ["streamlit", "run", "app.py", "--server.port", "8501", "--server.address", "0.0.0.0"]