floorplan-vectorizer / Dockerfile
Dharini Baskaran
initial commit
1d64201
raw
history blame
869 Bytes
# Use Python base image
FROM python:3.10-slim
# Set working directory
WORKDIR /app
# Install system dependencies
RUN apt-get update && apt-get install -y \
git \
build-essential \
libgl1-mesa-glx \
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
# Install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# Clone and install Detectron2
RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
RUN git clone https://github.com/facebookresearch/detectron2.git && \
pip install -e detectron2
# Copy your app code
COPY . .
# Expose Streamlit default port
EXPOSE 7860
# Streamlit Environment Variables
ENV STREAMLIT_SERVER_PORT=7860
ENV STREAMLIT_SERVER_HEADLESS=true
ENV STREAMLIT_SERVER_ENABLECORS=false
# Command to run the app
CMD ["streamlit", "run", "app.py"]