fix

Dockerfile

@@ -1,35 +0,0 @@
-# Use a lightweight Python base image
-FROM python:3.10-slim
-
-# Set environment variables
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONUNBUFFERED=1
-
-# Set working directory
-WORKDIR /app
-
-# Install system dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    git \
-    build-essential \
-    && rm -rf /var/lib/apt/lists/*
-
-# Install PyTorch CPU version
-RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-
-# Copy the segment-anything-2 repo and install it in editable mode
-COPY segment-anything-2 ./segment-anything-2
-RUN pip install -e ./segment-anything-2
-
-# Copy the manual requirements and install them without dependencies
-COPY requirements_manual.txt .
-RUN pip install --no-deps -r requirements_manual.txt
-
-# Copy rest of the app
-COPY . .
-
-# Expose Gradio/Streamlit default port
-EXPOSE 7860
-
-# Start the app (adjust depending on whether you're using Gradio or Streamlit)
-CMD ["python", "app.py"]

app.py

@@ -1,3 +1,24 @@
+import subprocess
+
+def setup_deps():
+    print("Setting up dependencies...")
+    try:
+        import torch
+        # Already installed? Skip setup
+        return
+    except ImportError:
+        pass
+
+    def run(cmd, cwd=None):
+        print(f"▶ {cmd}")
+        subprocess.check_call(cmd, shell=True, cwd=cwd)
+
+    run("pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu")
+    run("pip install -e .", cwd="segment-anything-2")
+    run("pip install --no-deps -r requirements_manual.txt")
+
+setup_deps()
+
 import gradio as gr
 import numpy as np
 from PIL import Image
@@ -7,6 +28,7 @@ from io import BytesIO
 
 from sam2.sam2_image_predictor import SAM2ImagePredictor
 
+
 # Dummy placeholders for SAM2 functions (replace with real logic)
 def segment_reference(image, click):
     # click = [x, y]