Update Dockerfile

Dockerfile

@@ -1,17 +1,11 @@
-# -------------------------------
 # Base image
-# -------------------------------
 FROM python:3.11-slim
 
-# -------------------------------
 # CPU-only environment
-# -------------------------------
 ENV TORCH_CUDA_ARCH_LIST=""
 ENV CUDA_VISIBLE_DEVICES=""
 
-# -------------------------------
 # System dependencies
-# -------------------------------
 RUN apt-get update && apt-get install -y \
     git \
     ffmpeg \
@@ -22,33 +16,23 @@ RUN apt-get update && apt-get install -y \
     libxrender-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# -------------------------------
 # Working directory
-# -------------------------------
 WORKDIR /app
 
-# -------------------------------
-# Copy requirements.txt
-# -------------------------------
+# Copy requirements
 COPY requirements.txt .
 
-# -------------------------------
-# Install CPU PyTorch first
-# -------------------------------
+# Install CPU PyTorch
 RUN pip install --no-cache-dir \
     torch==2.1.0 \
     torchvision==0.16.0 \
     torchaudio==2.1.0 \
     --index-url https://download.pytorch.org/whl/cpu
 
-# -------------------------------
-# Install compatible huggingface-hub first
-# -------------------------------
+# Install compatible huggingface-hub
 RUN pip install --no-cache-dir huggingface-hub>=0.19.3
 
-# -------------------------------
-# Install remaining Python packages
-# -------------------------------
+# Install remaining packages
 RUN pip install --no-cache-dir \
     git+https://github.com/ultralytics/ultralytics.git@main \
     transformers>=4.35.0 \
@@ -56,17 +40,11 @@ RUN pip install --no-cache-dir \
     gradio>=4.41.0 \
     opencv-python-headless
 
-# -------------------------------
-# Copy app code (optional, can add later)
-# -------------------------------
-# COPY . .
+# Copy app code
+COPY app.py .
 
-# -------------------------------
 # Expose Gradio port
-# -------------------------------
 EXPOSE 7860
 
-# -------------------------------
-# Default command (start bash for now)
-# -------------------------------
-CMD ["bash"]
+# Run the app
+CMD ["python", "app.py"]