Update Dockerfile

Dockerfile

@@ -1,7 +1,8 @@
 # Use the official Python 3.10 image as the base
+# Use the official Python 3.10 image as the base
 FROM python:3.10
 
-# Install system dependencies required for Hugging Face models and image processing
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     git-lfs \
@@ -13,21 +14,23 @@ RUN apt-get update && apt-get install -y \
     libgl1-mesa-glx \
     && rm -rf /var/lib/apt/lists/*
 
-# Set up Git LFS for model downloads
+# Set up Git LFS
 RUN git lfs install
 
-# Set the working directory
+# Set working directory
 WORKDIR /home/user/app
 
-# Copy the requirements file and install Python dependencies
-COPY requirements.txt /home/user/app/
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
+# Copy requirements file and install dependencies
+COPY requirements.txt /tmp/requirements.txt
+
+# Upgrade pip, setuptools, and wheel
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    pip install --no-cache-dir -r /tmp/requirements.txt
 
 # Copy application code into the container
 COPY . .
 
-# Expose Streamlit's default port (8501) if you're using Streamlit
+# Expose Streamlit's default port
 EXPOSE 8501
 
 # Command to run the app