Update Dockerfile

Dockerfile

@@ -1,16 +1,14 @@
-# Use Python 3.10-slim as base image
 FROM python:3.10-slim
 
-# Install system dependencies for OpenCV, FFmpeg, wget, and debugging
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     libgl1-mesa-glx \
     libglib2.0-0 \
     ffmpeg \
-    curl \
     wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Create necessary directories with proper permissions
+# Create directories
 RUN mkdir -p /tmp/Ultralytics /tmp/snapshots && \
     chmod -R 777 /tmp/Ultralytics /tmp/snapshots
 
@@ -24,10 +22,10 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY app.py /app.py
-RUN chmod 644 /app.py
+COPY data.yaml /data.yaml
 
-# Expose port
-EXPOSE 7860
+# Download custom model weights
+RUN wget -O /tmp/yolov8_custom.pt https://huggingface.co/PrashanthB461/SafetyViolationAI1/resolve/main/yolov8_custom.pt
 
-# Run app.py directly
+# Run the application
 CMD ["python", "/app.py"]