Update Dockerfile

Dockerfile

@@ -1,27 +1,20 @@
-# Use an official Python runtime as a parent image
-FROM python:3.9-slim
-
-# Set environment variables for Hugging Face cache
-ENV TRANSFORMERS_CACHE=/app/.cache
-ENV HF_DATASETS_CACHE=/app/.cache
-
-# Create the cache directory
-RUN mkdir -p /app/.cache
+# Stage 1: Build
+FROM python:3.9-slim as builder
 
-# Set the working directory in the container
 WORKDIR /app
-
-# Copy the requirements file into the container
 COPY requirements.txt .
+RUN pip install --no-cache-dir --user -r requirements.txt
 
-# Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+# Stage 2: Final Image
+FROM python:3.9-slim
+WORKDIR /app
 
-# Copy the current directory contents into the container
+# Copy only the necessary files from the builder stage
+COPY --from=builder /root/.local /root/.local
 COPY . .
 
-# Pre-download the model during the build process
-RUN python download_model.py
+# Ensure scripts in .local are usable
+ENV PATH=/root/.local/bin:$PATH
 
 # Expose port 7860 (default port for Gradio)
 EXPOSE 7860