Update Dockerfile

Dockerfile

@@ -6,7 +6,6 @@ RUN apt-get update -q && \
     apt-get install -y --no-install-recommends \
     python3.9 \
     python3-pip \
-    python-is-python3 \  # Ensures 'python' points to 'python3'
     libgl1 \
     libglib2.0-0 \
     wget \
@@ -23,27 +22,25 @@ RUN wget -qO /tmp/cuda-keyring.deb \
     libcublas-12-2=12.2.5.6-1 \
     && rm -rf /var/lib/apt/lists/*
 
-# Set working directory
 WORKDIR /app
 
-# Copy requirements and install Python dependencies
+# Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Create directories for models and temporary files
-RUN mkdir -p /app/weights/icon_detect /app/weights/icon_caption_florence /app/imgs \
-    && chmod -R 777 /app  # Ensure proper permissions
+# Create directories
+RUN mkdir -p /app/{weights,imgs} && \
+    chmod -R 777 /app
 
-# Copy application code
+# Application code
 COPY main.py utils.py download_models.py ./
 
-# Download models during build
-RUN python3 download_models.py  # Use python3 explicitly
+# Download models
+RUN python3 download_models.py
 
-# Set up non-root user for Hugging Face Spaces
+# Set up user
 RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
-# Run the application
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]