Update Dockerfile

Dockerfile

@@ -1,5 +1,6 @@
 FROM python:3.10-slim
 
+# Set working directory
 WORKDIR /app
 
 # Install system dependencies
@@ -9,28 +10,26 @@ RUN apt-get update && \
         wget \
         curl \
         ca-certificates \
-        gnupg \
         lsb-release \
-        unzip \
+        gnupg \
     && rm -rf /var/lib/apt/lists/*
 
-# Upgrade pip
-RUN python -m pip install --upgrade pip
-
 # Install Python packages
-RUN pip install --no-cache-dir torch transformers accelerate huggingface_hub git-lfs
-RUN pip install --no-cache-dir -U "huggingface_hub[cli]"
+RUN pip install --upgrade pip
+RUN pip install --no-cache-dir torch transformers accelerate huggingface_hub gradio
 
-# Copy requirements and install
+# Copy requirements first for caching
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy app code
+# Copy the rest of the app code
 COPY . /app
 
 # Expose port
 EXPOSE 7860
 
+# Run app
 CMD ["python", "app.py"]
 
 
+