Update Dockerfile

modified Dockerfile and replace any reference to /root/.cache with /tmp/huggingface_cache

Dockerfile

@@ -1,44 +1,38 @@
-# Use an official Python runtime as the base image
+# Use an official Python runtime as a parent image
 FROM python:3.9-slim
 
-# Set the working directory inside the container
+# Set the working directory to /app
 WORKDIR /app
 
-# Install system dependencies for ML and data processing libraries
+# Install system dependencies for machine learning and data processing libraries
 RUN apt-get update && apt-get install -y \
     build-essential \
     libopenblas-dev \
     libomp-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Upgrade pip to avoid dependency issues
-RUN pip install --upgrade pip
+# Create writable cache directories for Hugging Face and Matplotlib
+RUN mkdir -p /tmp/huggingface_cache /tmp/matplotlib
 
-# Set writable cache directories for Hugging Face and Matplotlib
-RUN mkdir -p /root/.cache/huggingface /tmp/matplotlib
-
-# Set environment variables
-ENV HF_HOME=/root/.cache/huggingface
+#  Set environment variables
+ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache
+ENV HF_HOME=/tmp/huggingface_cache
 ENV MPLCONFIGDIR=/tmp/matplotlib
 
-
-# Add Hugging Face API token (replace YOUR_HF_TOKEN with your actual token)
-ENV HUGGINGFACE_TOKEN=${HUGGINGFACE_TOKEN}
-
-
-
-
-# Copy the dependencies file first for caching efficiency
+# Copy requirements.txt first to leverage Docker caching
 COPY requirements.txt /app/requirements.txt
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the application code
+# Install PyTorch separately to ensure compatibility with the correct architecture
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
+#  Copy the rest of the application code
 COPY . /app
 
-# Expose port 7860 (required by Hugging Face Spaces)
+# Expose port 7860 for Hugging Face Spaces
 EXPOSE 7860
 
-# Command to run the Flask app
+#  Command to run your Flask app
 CMD ["python", "main.py"]