Update Dockerfile

Dockerfile

@@ -6,6 +6,7 @@ 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 \
@@ -22,25 +23,27 @@ 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
 
-# Python dependencies
+# Copy requirements and install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Create directories
-RUN mkdir -p /app/{weights,imgs} && \
-    chmod -R 777 /app
+# 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
 
-# Application code
+# Copy application code
 COPY main.py utils.py download_models.py ./
 
-# Download models
-RUN python download_models.py
+# Download models during build
+RUN python3 download_models.py  # Use python3 explicitly
 
-# Set up user
+# Set up non-root user for Hugging Face Spaces
 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"]