deployment using docker

app.py

@@ -172,4 +172,4 @@ def predict():
 
 if __name__ == "__main__":
     print(f"Running on: {device}")
-    app.run(debug=True, port=5000)
+    app.run(debug=True, port=7860)

deploy.dockerfile

@@ -0,0 +1,27 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# System dependencies
+RUN apt-get update && apt-get install -y \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install CPU-only PyTorch first (keeps image small)
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+
+# Install rest of dependencies
+COPY requirements.txt .
+RUN pip install -r requirements.txt
+
+# Copy project files
+COPY . .
+
+# HuggingFace Spaces requires port 7860
+EXPOSE 7860
+
+CMD ["python", "app.py"]

requirements.txt

@@ -1,8 +1,6 @@
 flask
-torch
-torchvision
 segmentation-models-pytorch
 albumentations
-opencv-python
+opencv-python-headless
 pillow
-numpy
+numpy