update dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-# Use an official Python runtime as a parent image
+# Use an official Python runtime
 FROM python:3.10-slim
 
 # Set environment variables
@@ -6,30 +6,34 @@ ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 ENV HOME=/home/user
 
-# Install system dependencies for OpenCV and FFmpeg
+# Install system dependencies
+# We use libgl1 instead of libgl1-mesa-glx for newer Debian compatibility
 RUN apt-get update && apt-get install -y \
-    libgl1-mesa-glx \
+    libgl1 \
     libglib2.0-0 \
     ffmpeg \
+    procps \
     && rm -rf /var/lib/apt/lists/*
 
-# Create a non-root user for Hugging Face
+# Create a non-root user (Required by Hugging Face)
 RUN useradd -m -u 1000 user
 USER user
+ENV PATH="/home/user/.local/bin:${PATH}"
 WORKDIR $HOME/app
 
 # Copy requirements and install
 COPY --chown=user requirements.txt .
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
 
 # Copy the rest of the application code
 COPY --chown=user . .
 
-# Create necessary directories and set permissions
+# Create storage directories and ensure they are writable
 RUN mkdir -p uploads outputs && chmod 777 uploads outputs
 
-# Expose port 7860 (Required by Hugging Face)
+# Expose the port HF expects
 EXPOSE 7860
 
-# Run the application
+# Start the FastAPI app
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -12,6 +12,16 @@ from ultralytics import YOLO
 
 app = FastAPI()
 
+
+# Create absolute paths based on the app directory
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
+OUTPUT_DIR = os.path.join(BASE_DIR, "outputs")
+
+# Ensure they exist (as a backup)
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+os.makedirs(OUTPUT_DIR, exist_ok=True)
+
 # --- CONFIG & GLOBALS ---
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 MODEL_PATH = "best_slowfast_theft.pth"