Update Dockerfile

Dockerfile

@@ -1,30 +1,37 @@
+# Base image
 FROM python:3.10-slim
 
-WORKDIR /app
+# Set env to avoid interactive prompts and ensure models cache to /tmp
+ENV HF_HOME=/tmp \
+    XDG_CACHE_HOME=/tmp \
+    TRANSFORMERS_CACHE=/tmp \
+    PYTHONUNBUFFERED=1
 
 # Install system dependencies
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends \
+RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/*
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libglib2.0-0 \
+    libblas-dev \
+    liblapack-dev \
+    gfortran \
+ && apt-get clean \
+ && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
+WORKDIR /app
 
-# Copy requirements and install dependencies
+# Copy requirements first for better caching
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
+# Copy the application code
 COPY . .
 
-# Set environment variables
-ENV PYTHONUNBUFFERED=1
-
-# Create cache directory with proper permissions
-RUN mkdir -p /root/.cache/huggingface && \
-    chmod -R 777 /root/.cache/huggingface
-
-# Expose port for the service
+# Expose port (for local or HF Spaces)
 EXPOSE 7860
 
-# Command to run the application
+# Start the Flask app
 CMD ["python", "app.py"]